only small datasets
This commit is contained in:
parent
f69eb59eb8
commit
5f871feff8
|
@ -165,7 +165,7 @@ if __name__ == '__main__':
|
||||||
np.random.seed(0)
|
np.random.seed(0)
|
||||||
|
|
||||||
optim_losses = ['mae'] # ['mae', 'mrae']
|
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()
|
models = quantification_models()
|
||||||
|
|
||||||
results = Parallel(n_jobs=settings.N_JOBS)(
|
results = Parallel(n_jobs=settings.N_JOBS)(
|
||||||
|
|
Loading…
Reference in New Issue