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)