如何使用keras.js进行预测

时间:2018-12-15 20:56:33

标签: javascript reactjs keras

我是keras.js的新手,并使用Keras.JS对输入进行了预测 我正在使用以下代码加载和训练模型

要加载JSON文件:

async load(event)
    {
      const response = await fetch('./model.json');
      const json = await response;
      this.setState({ value: json });
      this.predictedValue = this.predictValue([this.state.value]);
    }

要预测:

 predictValue(inputs) 
   { 
      const res = new Float32Array(inputs);
      const prediction = KerasJS.model.predict(res);
      return prediction.get(0, 0);
    }

但是输出为[object Response]。如何获得预测值?

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

由于predict返回了一个承诺(根据docs),您应该使用其中任何一个

const prediction = await KerasJS.model.predict(inputData);

thencatch方法

KerasJS.model.predict(res)
.then(prediction => {
// prediction is an object keyed by names of the output layers
// or `output` for Sequential models
// e.g.,
// prediction ['fc1000']
})
.catch(err => {
// handle error
})

希望这会有所帮助!