def test_rocket_on_gunpoint():
# load gunpoint data
X_train, y_train = load_gunpoint(split="train", return_X_y=True)
X_test, y_test = load_gunpoint(split="test", return_X_y=True)
indices = np.random.RandomState(0).permutation(10)
# train ROCKET
rocket = ROCKETClassifier(num_kernels=1000, random_state=0)
rocket.fit(X_train.iloc[indices], y_train[indices])
# assert probabilities are the same
probas = rocket.predict_proba(X_test.iloc[indices])
testing.assert_array_equal(probas, rocket_gunpoint_probas)
def test_rocket_on_basic_motions():
"""Test of ROCKET on basic motions."""
# load basic motions data
X_train, y_train = load_basic_motions(split="train", return_X_y=True)
X_test, y_test = load_basic_motions(split="test", return_X_y=True)
indices = np.random.RandomState(0).permutation(20)
# train ROCKET
rocket = ROCKETClassifier(num_kernels=1000, random_state=0)
rocket.fit(X_train.iloc[indices], y_train[indices])
# assert probabilities are the same
probas = rocket.predict_proba(X_test.iloc[indices])
testing.assert_array_equal(probas, rocket_basic_motions_probas)
def test_rocket_on_unit_test_data():
"""Test of ROCKETClassifier on unit test data."""
# load unit test data
X_train, y_train = load_unit_test(split="train", return_X_y=True)
X_test, y_test = load_unit_test(split="test", return_X_y=True)
indices = np.random.RandomState(0).choice(len(y_train), 10, replace=False)
# train ROCKET
rocket = ROCKETClassifier(num_kernels=500, random_state=0)
rocket.fit(X_train, y_train)
# assert probabilities are the same
probas = rocket.predict_proba(X_test.iloc[indices])
testing.assert_array_almost_equal(probas,
rocket_unit_test_probas,
decimal=2)