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
@ -164,8 +164,8 @@ if __name__ == '__main__':
    print(f'Result folder: {args.results}')
    np.random.seed(0)

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

    results = Parallel(n_jobs=settings.N_JOBS)(