我用npm安装了@ tensorflow-models / knn-classifier,但是运行时我收到一个错误“找不到模块'@ tensorflow-models / knn-classifier'”。
我可以在节点模块下看到该模块,但仍然出现此错误。其他模型(如@ tensorflow-models / mobilenet,@ tensorflow-models / universal-sentence-encoder)可以解析,但knn-classifier不能解析。此模型是否还需要其他东西?
import * as tf from '@tensorflow/tfjs';
import * as knnClassifier from '@tensorflow-models/knn-classifier';
const classifier = knnClassifier.create();
const trainData = getTrainingData();
for(const td of trainData) {
classifier.addExample(td.xs, td.label);
}
最佳答案
目前,该软件包缺少index.js
。这是解决方法
const tf = require('@tensorflow/tfjs');
const knnClassifier = require('./node_modules/@tensorflow-models/knn-classifier/dist/knn-classifier');
const classifier = knnClassifier.create();
console.log('classifier', classifier)