only small datasets

2021-01-20 09:05:22 +01:00 · 2021-01-20 09:05:22 +01:00 · 5f871feff8
parent f69eb59eb8
commit 5f871feff8
1 changed files with 2 additions and 2 deletions
--- a/TweetSentQuant/experiments.py
+++ b/TweetSentQuant/experiments.py
@ -165,7 +165,7 @@ if __name__ == '__main__':
    np.random.seed(0)

    optim_losses = ['mae'] # ['mae', 'mrae']
-    datasets = qp.datasets.TWITTER_SENTIMENT_DATASETS_TRAIN
+    datasets = ['hcr', 'omd', 'sanders', 'sst'] # qp.datasets.TWITTER_SENTIMENT_DATASETS_TRAIN
    models = quantification_models()

    results = Parallel(n_jobs=settings.N_JOBS)(