ソースを参照

Support CoreML support vector regressor

Lutz Roeder 8 年 前
コミット
3d037a519d
1 ファイル変更10 行追加0 行削除
  1. 10 0
      src/coreml-model.js

+ 10 - 0
src/coreml-model.js

@@ -264,6 +264,16 @@ class CoreMLGraph {
             this.updateClassifierOutput(group, model.supportVectorClassifier);
             return 'Support Vector Classifier';            
         }
+        else if (model.supportVectorRegressor) {
+            this._nodes.push(new CoreMLNode(group, 'supportVectorRegressor', null, 
+                { coefficients: model.supportVectorRegressor.coefficients, 
+                  kernel: model.supportVectorRegressor.kernel,
+                  rho: model.supportVectorRegressor.rho,
+                  supportVectors: model.supportVectorRegressor.supportVectors }, 
+                [ model.description.input[0].name ],
+                [ model.description.output[0].name ]));
+            return 'Support Vector Regressor';
+        }
         return 'Unknown';
     }