Tensorflow.js入门之toxicity

最新推荐文章于 2022-07-04 17:44:30 发布

Maann

最新推荐文章于 2022-07-04 17:44:30 发布

阅读量346

点赞数

分类专栏：深度学习文章标签： tensorflow 深度学习 python javascript html

本文链接：https://blog.csdn.net/weixin_43815222/article/details/108260976

版权

深度学习专栏收录该内容

20 篇文章 3 订阅

订阅专栏

tfjs入门之toxicity

教程：【吴恩达团队Tensorflow2.0数据和部署第一课】基于浏览器的模型与TensorFlow.js

toxicity.html

<html>
<head>
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.1.0"></script>
<script src="https://cdn.jsdelivr.net/npm/@tensorflow-models/toxicity@1.2.0"></script>
<script>
// The minimum prediction confidence.
const threshold = 0.9;
// Load the model. Users optionally pass in a threshold and an array of
// labels to include.
toxicity.load(threshold).then(model => {
  const sentences = ['you suck'];
  model.classify(sentences).then(predictions => {
    console.log(predictions);
	for(i=0;i<7;i++){
	    if(predictions[i].results[0].match){
	        console.log(predictions[i].label *" was found with a probability of " * predictions[i].results[0].probabilities[1]);
		}
	}

  });
});
</script>
</head>
<body></body>
</html>

在浏览器打开，按F12
在这里插入图片描述
则成功

说一说遇到的坑：
（1）指定tfjs和toxicity版本问题，此处分别为2.1.0，1.2.0，否则可能会遇到找不到函数等各种bug

（2）警告信息可以忽略，或者脚本标签改为
这里是引用

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.1.0/dist/tf.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/@tensorflow-models/toxicity@1.2.0/dist/toxicity.js"></script>