Explorar o código

1- 去掉TODO注释。

bijl %!s(int64=6) %!d(string=hai) anos
pai
achega
30a4c5ef2f

+ 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(