🔧 Various fixes

This commit is contained in:
2021-09-08 22:52:00 -07:00
parent f3d94f2a75
commit f04641986d
10 changed files with 145 additions and 24 deletions

View File

@@ -0,0 +1,34 @@
import Classifier from "ml-classify-text";
export const detectTradePaperbacks = (deck: string): any => {
const classifier = new Classifier({ nGramMin: 2, nGramMax: 2 });
const positiveTPBIdentifiers = [
"trade paperbacks",
"TPB",
"paperback",
"hardcover",
"collects the following issues",
"collected issues",
"collecting the issues",
"collecting the following issues",
"collected editions",
];
const negativeTPBIdentifiers = ["mini-series"];
classifier.train(positiveTPBIdentifiers, "Possibly a trade paperback");
classifier.train(negativeTPBIdentifiers, "Not a trade paperback");
if (deck) {
console.log("DEC", deck);
const predictions = classifier.predict(deck);
if (predictions.length) {
predictions.forEach((prediction) => {
console.log(`${prediction.label} (${prediction.confidence})`);
return prediction;
});
} else {
console.log("No predictions returned.");
}
}
};

View File

@@ -0,0 +1,14 @@
[
{"phrase": "Twelve issue mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Trade paperback collecting Jean Grey #7-11.", "result": {"paperback": 1}},
{"phrase": "Trade paperback collecting Iron Fist #6-7 & 73-77", "result": {"paperback ": 1}},
{"phrase": "Collected Editions", "result": {"paperback": 1}},
{"phrase": "Six issue mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Four issue mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Seventy-two issue digital prequel to the Injustice 2 video game.", "result": {"not_a_paperback": 1}},
{"phrase": "Total issues in this volume:50", "result": {"paperback": 1}},
{"phrase": "Five issue mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Digital mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Six issue mini-series.", "result": {"not_a_paperback": 1}},
{"phrase": "Six issue mini-series.", "result": {"not_a_paperback": 1}}
]