def test_extra_trees_regressor(self):
     model = ExtraTreesRegressor(n_estimators=3)
     dump_single_regression(
         model,
         allow_failure=
         "StrictVersion(onnxruntime.__version__) <= StrictVersion('0.2.1')")
     dump_multiple_regression(
         model,
         allow_failure=
         "StrictVersion(onnxruntime.__version__) <= StrictVersion('0.2.1')")
 def test_decision_tree_regressor(self):
     model = DecisionTreeRegressor()
     dump_single_regression(
         model,
         allow_failure=
         "StrictVersion(onnx.__version__) < StrictVersion('1.2')")
     dump_multiple_regression(
         model,
         allow_failure=
         "StrictVersion(onnx.__version__) < StrictVersion('1.2')")
 def test_random_forest_regressor(self):
     model = RandomForestRegressor(n_estimators=3)
     dump_single_regression(
         model,
         allow_failure=
         "StrictVersion(onnxruntime.__version__) <= StrictVersion('0.2.1')")
     dump_multiple_regression(
         model,
         allow_failure=
         "StrictVersion(onnxruntime.__version__) <= StrictVersion('0.2.1')")
Beispiel #4
0
 def test_extra_tree_regressor(self):
     model = ExtraTreeRegressor()
     dump_single_regression(
         model,
         allow_failure="StrictVersion(onnx.__version__)"
                       " < StrictVersion('1.2')",
     )
     dump_multiple_regression(
         model,
         allow_failure="StrictVersion(onnx.__version__)"
                       " < StrictVersion('1.2')",
     )
Beispiel #5
0
 def test_extra_trees_regressor(self):
     model = ExtraTreesRegressor(n_estimators=3)
     dump_single_regression(model)
     dump_multiple_regression(model)
Beispiel #6
0
 def test_random_forest_regressor(self):
     model = RandomForestRegressor(n_estimators=3)
     dump_single_regression(model)
     dump_multiple_regression(model)
Beispiel #7
0
 def test_extra_tree_regressor(self):
     model = ExtraTreeRegressor()
     dump_single_regression(model)
     dump_multiple_regression(model)
Beispiel #8
0
 def test_decision_tree_regressor(self):
     model = DecisionTreeRegressor()
     dump_single_regression(model)
     dump_multiple_regression(model)