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')" ), )
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)
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')")
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)
def test_extra_tree_classifier(self): model = ExtraTreeClassifier() dump_one_class_classification(model) dump_binary_classification(model) dump_multiple_classification(model)