66c20cc2157d7f18530c742ce6d71685f8699020,tests/classifiers/test_xgboost.py,TestXGBoostClassifierPythonAPI,setUpClass,#Any#,89

Before Change

    @classmethod
    def setUpClass(cls):
        np.random.seed(seed=1234)
        (x_train, y_train), (x_test, y_test), _, _ = load_dataset("iris")

        cls.x_train = x_train
        cls.y_train = y_train

After Change

    @classmethod
    def setUpClass(cls):
        super().setUpClass()

        cls.y_test_iris = np.argmax(cls.y_test_iris, axis=1)

        model = xgb.XGBClassifier(n_estimators=30, max_depth=5)

Instances