Browse Source

Merge remote-tracking branch 'origin/master'

louhr 6 years atrás
parent
commit
6c487f9d7e

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