def test_predict_batched_sparse(self):
dataset_properties = {'sparse': True}
include = {'regressor': ['decision_tree']}
cs = SimpleRegressionPipeline(
dataset_properties=dataset_properties,
include=include).get_hyperparameter_search_space()
default = cs.get_default_configuration()
regressor = SimpleRegressionPipeline(
config=default,
random_state=1,
dataset_properties=dataset_properties,
include=include)
X_train, Y_train, X_test, Y_test = get_dataset(dataset='boston',
make_sparse=True)
regressor.fit(X_train, Y_train)
X_test_ = X_test.copy()
prediction_ = regressor.predict(X_test_)
mock_predict = unittest.mock.Mock(
wraps=regressor.steps[-1][-1].predict)
regressor.steps[-1][-1].predict = mock_predict
prediction = regressor.predict(X_test, batch_size=20)
self.assertEqual((356, ), prediction.shape)
self.assertEqual(18, mock_predict.call_count)
np.testing.assert_array_almost_equal(prediction_, prediction)
def test_predict_batched(self):
include = {'regressor': ['decision_tree']}
cs = SimpleRegressionPipeline(include=include).get_hyperparameter_search_space()
default = cs.get_default_configuration()
regressor = SimpleRegressionPipeline(default, include=include)
X_train, Y_train, X_test, Y_test = get_dataset(dataset='boston')
regressor.fit(X_train, Y_train)
X_test_ = X_test.copy()
prediction_ = regressor.predict(X_test_)
mock_predict = unittest.mock.Mock(wraps=regressor.steps[-1][-1].predict)
regressor.steps[-1][-1].predict = mock_predict
prediction = regressor.predict(X_test, batch_size=20)
self.assertEqual((356,), prediction.shape)
self.assertEqual(18, mock_predict.call_count)
assert_array_almost_equal(prediction_, prediction)
def test_predict_batched(self):
include = {'regressor': ['decision_tree']}
cs = SimpleRegressionPipeline(include=include).get_hyperparameter_search_space()
default = cs.get_default_configuration()
regressor = SimpleRegressionPipeline(default, include=include)
X_train, Y_train, X_test, Y_test = get_dataset(dataset='boston')
regressor.fit(X_train, Y_train)
X_test_ = X_test.copy()
prediction_ = regressor.predict(X_test_)
mock_predict = unittest.mock.Mock(wraps=regressor.steps[-1][-1].predict)
regressor.steps[-1][-1].predict = mock_predict
prediction = regressor.predict(X_test, batch_size=20)
self.assertEqual((356,), prediction.shape)
self.assertEqual(18, mock_predict.call_count)
assert_array_almost_equal(prediction_, prediction)
