Skip to content

Instantly share code, notes, and snippets.

@martzcodes
Created May 5, 2019 16:15
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save martzcodes/79f4f88480721f75f3d5f69d36d59801 to your computer and use it in GitHub Desktop.
Save martzcodes/79f4f88480721f75f3d5f69d36d59801 to your computer and use it in GitHub Desktop.
import * as toxicity from "@tensorflow-models/toxicity";
// ...
toxicity.load(threshold, []).then(model => {
console.log("toxicity model loaded");
this.toxicityModel = model;
});
// ...
return this.toxicityModel.classify(textToAnalyze).then(predictions => {
console.log("Predictions:", predictions);
this.predictions$.next(predictions);
this.analyzing$.next(false);
});
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment