def test_super_learner_f_t(): """[SuperLearner] Test scikit-learn comp - th | p""" ens = get_ensemble(SuperLearner, 'threading', prep) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)
def test_super_learner_s_m(): """[SuperLearner] Test scikit-learn comp - mp | np""" ens = get_ensemble(SuperLearner, 'multiprocessing', None) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)
def test_temporal_f_m(): """[TemporalEnsemble] Test scikit-learn comp - th | p""" ens = get_ensemble(TemporalEnsemble, 'threading', prep, step_size=10) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)
def test_blend_f_m(): """[BlendEnsemble] Test scikit-learn comp - th | p""" ens = get_ensemble(BlendEnsemble, 'threading', prep) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)
def test_subsemble_s_t(): """[Subsemble] Test scikit-learn comp - th | np""" ens = get_ensemble(Subsemble, 'threading', None) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)
def test_temporal_s_m(): """[TemporalEnsemble] Test scikit-learn comp - mp | np""" ens = get_ensemble(TemporalEnsemble, 'multiprocessing', None, step_size=10) ens.fit(X, y) p = ens.predict(X) assert p.shape == y.shape assert p.dtype == ens.layer_1.dtype ens = return_pickled(ens) pp = ens.predict(X) np.testing.assert_array_equal(p, pp)