Sentiment

Sentiment

Description

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 magic = 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.