UNPKG

ml5-save

Version:
112 lines (66 loc) • 2.39 kB
# Sentiment <center> <img style="display:block; max-height:20rem" alt="illustration of hand drawing a heart" src="_media/reference__header-sentiment.png"> </center> ## 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 ```js // 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 ```js 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. ### Properties *** #### .ready > Boolean value that specifies if the model has loaded. *** *** #### .model > The model being used. *** ### Methods *** #### .predict() > Given a number, will make magicSparkles ```js sentiment.predict(text); ``` šŸ“„ **Inputs** * **text**: Required. String. A string of text to predict šŸ“¤ **Outputs** * **Object**: Scores the sentiment of given text with a value between 0 ("negative") and 1 ("positive"). *** ## Examples **p5.js** * [Sentiment_Interactive](https://github.com/ml5js/ml5-examples/tree/development/p5js/Sentiment/Sentiment_Interactive) **p5 web editor** * [Sentiment_Interactive](https://editor.p5js.org/ml5/sketches/Sentiment_Interactive) **plain javascript** * [Sentiment_Interactive](https://github.com/ml5js/ml5-examples/tree/development/javascript/Sentiment/Sentiment_Interactive) ## Demo No demos yet - contribute one today! ## Tutorials No tutorials yet - contribute one today! ## Acknowledgements **Contributors**: * Itay Niv **Credits**: * Paper Reference | Website URL | Github Repo | Book reference | etc ## Source Code [/src/Sentiment/](https://github.com/ml5js/ml5-library/tree/development/src/Sentiment)