使用 npm 安装 @tensorflow-models/knn-classifier 问题

Question

我用 npm 安装了 @tensorflow-models/knn-classifier，但是当我运行它时，我得到一个错误 "Cannot find module '@tensorflow-models/knn-classifier'"。我可以在节点模块下看到该模块，但我仍然收到此错误。 @tensorflow-models/mobilenet、@tensorflow-models/universal-sentence-encoder 等其他模型已解析，但 knn 分类器未解析。这个模型需要额外的东西吗？

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);    
}

Answer 1

目前缺少包 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)

使用 npm 安装 @tensorflow-models/knn-classifier 问题

Issue installing @tensorflow-models/knn-classifier with npm

javascript

node.js

tensorflow.js