فهرست منبع

1- 去掉TODO注释。

bijl 6 سال پیش
والد
کامیت
30a4c5ef2f
1فایلهای تغییر یافته به همراه1 افزوده شده و 2 حذف شده
  1. 1 2
      algorithm/src/main/java/org/algorithm/core/neural/TensorflowModel.java

+ 1 - 2
algorithm/src/main/java/org/algorithm/core/neural/TensorflowModel.java

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