node-fasttext
v1.0.0-alpha.1
Published
A facebookresearch/fastText wrapper in node js.
Downloads
50
Maintainers
Readme
node-fasttext
npm install --save node-fasttext
To import: var FastText = require('node-fasttext');
Training
let config = {
dim: 100,
input: "data/dbpedia.train",
output: "dbpedia"
}
FastText.train("supervised", config, function (success, error) {
if(error) {
console.log(error)
return;
}
console.log(success)
})
Print Vector
FastText.printVectors(
"dbpedia.bin",
['word', 'word', 'another', 'words', 'here'],
function (success, error) {
if(error) {
console.log(error)
return;
}
console.log(success.length)
})
Print model info
FastText.modelInfo("dbpedia.bin", function (success, error) {
if(error) {
console.log(error)
return;
}
console.log(success)
})
Test a model
FastText.test("dbpedia.bin", "data/dbpedia.test", 1, function (data, error) {
if(error) {
console.log(error)
return;
}
console.log(data)
});
Predict
FastText.predict(
"dbpedia.bin", 1,
['cola lukei , cola lukei is a species of flowering plant in the malvaceae family . it is found only in tanzania . its natural habitat is subtropical or tropical moist lowland forests . it is threatened by habitat loss .'],
function (success, error) {
if(error) {
console.log(error)
return;
}
console.log(success[0].label == '__label__11')
console.log(success)
})