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)