Ejemplos de RayXGBClassifier.get_xgb_params en Python

Ejemplo n.º 1

Archivo: test_sklearn.py Proyecto: krfricke/xgboost_ray

    def test_sklearn_n_jobs(self):
        self._init_ray()

        clf = RayXGBClassifier(n_jobs=1)
        assert clf.get_xgb_params()["n_jobs"] == 1

        clf = RayXGBClassifier(n_jobs=2)
        assert clf.get_xgb_params()["n_jobs"] == 2

Ejemplo n.º 2

Archivo: test_sklearn.py Proyecto: krfricke/xgboost_ray

    def test_sklearn_random_state(self):
        self._init_ray()

        clf = RayXGBClassifier(random_state=402)
        assert clf.get_xgb_params()["random_state"] == 402

        clf = RayXGBClassifier(random_state=401)
        assert clf.get_xgb_params()["random_state"] == 401

        random_state = np.random.RandomState(seed=403)
        clf = RayXGBClassifier(random_state=random_state)
        assert isinstance(clf.get_xgb_params()["random_state"], int)