Skip to content

Commit

Permalink
commenting cross validation test for ssml since it will be included i…
Browse files Browse the repository at this point in the history
…n the LabelProp PR
  • Loading branch information
Jordan Stomps committed Oct 31, 2022
1 parent c661099 commit a7d4bfe
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,15 +67,15 @@ def test_utils():
stratified=True)
assert max_acc_model['accuracy'] >= 0.5

# test cross validation for SSML with LabelProp
params = {'gamma': 10, 'n_neighbors': 15, 'max_iter': 2022, 'tol': 0.5}
model = LabelProp(params=params)
max_acc_model = utils.cross_validation(model=model,
X=np.append(X, Ux, axis=0),
y=np.append(y, Uy, axis=0),
params=params,
stratified=True)
assert max_acc_model['accuracy'] >= 0.5
# # test cross validation for SSML with LabelProp
# params = {'gamma': 10, 'n_neighbors': 15, 'max_iter': 2022, 'tol': 0.5}
# model = LabelProp(params=params)
# max_acc_model = utils.cross_validation(model=model,
# X=np.append(X, Ux, axis=0),
# y=np.append(y, Uy, axis=0),
# params=params,
# stratified=True)
# assert max_acc_model['accuracy'] >= 0.5

# data split for data visualization
X_train, X_test, y_train, y_test = train_test_split(X,
Expand Down

0 comments on commit a7d4bfe

Please sign in to comment.