Sentiment is a model trained to predict the sentiment of any given text. The default model, currently 'moviereviews
', is trained using IMDB reviews that have been truncated to a maximum of 200 words, only the 20000 most used words in the reviews are used.
Quickstart :
// Create a new Sentiment method
const sentiment = ml5.sentiment('movieReviews', modelReady);
// When the model is loaded
function modelReady() {
// model is ready
console.log('Model Loaded!');
}
// make the prediction
const prediction = sentiment.predict(text);
console.log(prediction);
Usage :
Initialize
const sentiment = ml5.sentiment(model, ?callback);
Parameters
* model : REQUIRED. Defaults to 'moviereviews
'. You can also use a path to a manifest.json
file via a relative or absolute path.
* callback : OPTIONAL. A callback function that is called once the model has loaded. If no callback is provided, it will return a promise that will be resolved once the model has loaded.