diff --git a/database/data_ISA_aligned_prosit.csv b/database/data_ISA_aligned_prosit.csv index e26afd0fecff7f9a2d79ca3c188f489d474f1b7a..ad48f91637989d9241567186af0a688cf8e801ca 100644 Binary files a/database/data_ISA_aligned_prosit.csv and b/database/data_ISA_aligned_prosit.csv differ diff --git a/main.py b/main.py index 97cecf39b7b11529db2af847707808a58a76f273..936e040d24cb16b9c4fbacd8e64a550ea258a44e 100644 --- a/main.py +++ b/main.py @@ -272,7 +272,7 @@ def main_rt(args): data_sources=[args.dataset_train, args.dataset_train, args.dataset_train]) else: data_train, data_val, data_test = load_data(batch_size=args.batch_size, n_train=args.n_train, n_test=args.n_test, - data_sources=[args.dataset_train,args.dataset_train,args.dataset_test]) + data_sources=[args.dataset_train,args.dataset_test,args.dataset_train]) print('\nData loaded') # if args.model == 'RT_self_att' : # model = RT_pred_model_self_attention()