060a71ef947f9d9e87e94dd8320ef78c83c946cb,tests/xgboost/test_xgboost_converters.py,TestXGBoostModels,test_xgb_classifier_multi_reglog,#TestXGBoostModels#,53

Before Change

    @unittest.skipIf(sys.version_info[0] == 2, reason="xgboost converter not tested on python 2")
    def test_xgb_classifier_multi_reglog(self):
        iris = load_iris()
        X = iris.data[:, :2]
        y = iris.target

        xgb = XGBClassifier(objective="reg:logistic")
        xgb.fit(X, y)
        conv_model = convert_xgboost(xgb, initial_types=[("input", FloatTensorType(shape=[1, 2]))])
        self.assertTrue(conv_model is not None)
        dump_multiple_classification(xgb, suffix="RegLog",
                                     allow_failure="StrictVersion(onnx.__version__) < StrictVersion("1.3.0")")

After Change

    @unittest.skipIf(sys.version_info[0] == 2,
                     reason="xgboost converter not tested on python 2")
    def test_xgb_classifier_multi_reglog(self):
        xgb, x_test = _fit_classification_model(
            XGBClassifier(objective="reg:logistic"), 4)
        conv_model = convert_xgboost(
            xgb, initial_types=[("input", FloatTensorType(shape=[1, 2]))])
        self.assertTrue(conv_model is not None)
        dump_data_and_model(

Instances