|
@@ -59,12 +59,11 @@ public class TensorflowModel {
|
|
|
|
|
|
/**
|
|
/**
|
|
* 运行模型
|
|
* 运行模型
|
|
- * @param inputValue 输入值
|
|
|
|
|
|
+ * @param inputValues 输入值
|
|
* @param numExamples 样本个数
|
|
* @param numExamples 样本个数
|
|
* @return 模型的输出
|
|
* @return 模型的输出
|
|
*/
|
|
*/
|
|
private float[][] run(float[] inputValues, int numExamples){
|
|
private float[][] run(float[] inputValues, int numExamples){
|
|
- // TODO: 修改的地方
|
|
|
|
// long[] inputShape = {numExamples, this.NUM_FEATURE, 4, 1}; // 新模型
|
|
// long[] inputShape = {numExamples, this.NUM_FEATURE, 4, 1}; // 新模型
|
|
long[] inputShape = {numExamples, this.NUM_FEATURE}; // 老模型
|
|
long[] inputShape = {numExamples, this.NUM_FEATURE}; // 老模型
|
|
Tensor<Float> inputTensor = Tensor.create(
|
|
Tensor<Float> inputTensor = Tensor.create(
|