总字数 460
预计阅读时间 1 分钟
模型训练好之后,就可以使用该模型进行预测了
代码依然是nodejs环境的运行方式
1 | import * as tf from '@tensorflow/tfjs-node' |
这里在本地使用http-server
启动了一个HTTP服务,方便加载模型
上面代码最后输出的执行结果是
1 | [ |
与4种类型相对应
1 | ["其他垃圾","厨余垃圾","可回收物","有害垃圾"] |
显然与可回收物的匹配度较高,其他几种的匹配度较低
实际预测的结果与模型的设定以及训练的素材数量相关