Пример #1
0
 def test_extra_trees_classifier(self):
     model = ExtraTreesClassifier(n_estimators=3)
     dump_one_class_classification(
         model,
         allow_failure="StrictVersion(onnx.__version__)"
                       " < StrictVersion('1.2') or "
                       "StrictVersion(onnxruntime.__version__)"
                       " <= StrictVersion('0.2.1')",
     )
     dump_binary_classification(
         model,
         allow_failure=(
             "StrictVersion(onnx.__version__) < StrictVersion('1.2') or "
             "StrictVersion(onnxruntime.__version__)"
             " <= StrictVersion('0.2.1')"
         ),
     )
     dump_multiple_classification(
         model,
         # Operator cast-1 is not implemented in onnxruntime
         allow_failure=(
             "StrictVersion(onnx.__version__) < StrictVersion('1.2') or "
             "StrictVersion(onnxruntime.__version__)"
             " <= StrictVersion('0.2.1')"
         ),
     )
Пример #2
0
 def test_decision_tree_classifier(self):
     model = DecisionTreeClassifier()
     dump_one_class_classification(model)
     dump_binary_classification(model)
     dump_multiple_classification(model)
     dump_multiple_classification(model, label_uint8=True)
     dump_multiple_classification(model, label_string=True)
Пример #3
0
 def test_random_forest_classifier(self):
     model = RandomForestClassifier(n_estimators=3)
     dump_one_class_classification(model,
                                   allow_failure="StrictVersion(onnx.__version__) < StrictVersion('1.2')")
     dump_binary_classification(model,
                                allow_failure="StrictVersion(onnx.__version__) < StrictVersion('1.2')")
     dump_multiple_classification(model,
                                  allow_failure="StrictVersion(onnx.__version__) < StrictVersion('1.2')")
 def test_decision_tree_classifier(self):
     model = DecisionTreeClassifier()
     dump_one_class_classification(
         model,
         # Operator cast-1 is not implemented in onnxruntime
         allow_failure=
         "StrictVersion(onnx.__version__) < StrictVersion('1.2')")
     dump_binary_classification(
         model,
         allow_failure=
         "StrictVersion(onnx.__version__) < StrictVersion('1.2')")
     dump_multiple_classification(
         model,
         allow_failure=
         "StrictVersion(onnx.__version__) < StrictVersion('1.2')")
Пример #5
0
 def test_decision_tree_classifier(self):
     model = DecisionTreeClassifier()
     dump_one_class_classification(
         model,
         # Operator cast-1 is not implemented in onnxruntime
         allow_failure="StrictVersion(onnx.__version__)"
         " < StrictVersion('1.3') or "
         "StrictVersion(onnxruntime.__version__)"
         " <= StrictVersion('0.2.1')",
     )
     dump_binary_classification(
         model,
         allow_failure="StrictVersion(onnx.__version__)"
         " < StrictVersion('1.3') or "
         "StrictVersion(onnxruntime.__version__)"
         " <= StrictVersion('0.2.1')",
     )
     dump_multiple_classification(
         model,
         allow_failure="StrictVersion(onnx.__version__)"
         " < StrictVersion('1.3') or "
         "StrictVersion(onnxruntime.__version__)"
         " <= StrictVersion('0.2.1')")
     dump_multiple_classification(
         model,
         label_uint8=True,
         allow_failure="StrictVersion(onnx.__version__)"
         " < StrictVersion('1.3') or "
         "StrictVersion(onnxruntime.__version__)"
         " <= StrictVersion('0.2.1')")
     dump_multiple_classification(
         model,
         label_string=True,
         allow_failure="StrictVersion(onnx.__version__)"
         " < StrictVersion('1.3') or "
         "StrictVersion(onnxruntime.__version__)"
         " <= StrictVersion('0.2.1')")
 def test_random_forest_classifier(self):
     model = RandomForestClassifier(n_estimators=3)
     dump_one_class_classification(model)
     dump_binary_classification(model)
     dump_multiple_classification(model)
 def test_extra_trees_classifier(self):
     model = ExtraTreesClassifier(n_estimators=3)
     dump_one_class_classification(model)
     dump_binary_classification(model)
     dump_multiple_classification(model)
Пример #8
0
 def test_extra_tree_classifier(self):
     model = ExtraTreeClassifier()
     dump_one_class_classification(model)
     dump_binary_classification(model)
     dump_multiple_classification(model)