def test_weasel_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)
# train WEASEL
weasel = WEASEL(random_state=1, binning_strategy="equi-depth")
weasel.fit(X_train, y_train)
score = weasel.score(X_test, y_test)
# print(score)
assert score >= 0.99
def test_weasel_on_power_demand():
# load power demand data
X_train, y_train = load_italy_power_demand(split="train", return_X_y=True)
X_test, y_test = load_italy_power_demand(split="test", return_X_y=True)
# train WEASEL
weasel = WEASEL(random_state=1, binning_strategy="kmeans")
weasel.fit(X_train, y_train)
score = weasel.score(X_test, y_test)
# print(score)
assert score >= 0.94
def test_weasel_on_power_demand():
# load power demand data
X_train, y_train = load_italy_power_demand(split='train', return_X_y=True)
X_test, y_test = load_italy_power_demand(split='test', return_X_y=True)
# train WEASEL
weasel = WEASEL(random_state=47)
weasel.fit(X_train, y_train)
score = weasel.score(X_test, y_test)
print(score)
assert (score >= 0.94)
def test_weasel_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 WEASEL
weasel = WEASEL(random_state=1379)
weasel.fit(X_train, y_train)
score = weasel.score(X_test, y_test)
# print(score)
assert score >= 0.99