Merge remote-tracking branch 'upstream/main' into sentiment-analysis

Author: Lillliant

output/twitter/reviews.arb_Arab.pkl

@@ -255,6 +255,9 @@ def main(args):
 
     output = f'{output}/{am.name()}/'
 
+    eval_for: ModelCapabilities = set(params.settings['eval']['for']).intersection(am.capabilities) #type: ignore
+    train_for: ModelCapabilities = set(params.settings['train']['for']).intersection(am.capabilities) #type: ignore
+
     if 'train' in params.settings['cmd']:
         for capability in train_for:
             for f in splits['folds'].keys():
@@ -264,9 +267,6 @@ def main(args):
                 train(args, am, reviews_train, np.array(reviews)[splits['folds'][f]['valid']].tolist(), f, output, capability)
                 print(f'Trained time elapsed including language augs {params.settings["prep"]["langaug"]}: {time.time() - t_s}')
 
-    eval_for: ModelCapabilities = set(params.settings['eval']['for']).intersection(am.capabilities) #type: ignore
-    train_for: ModelCapabilities = set(params.settings['train']['for']).intersection(am.capabilities) #type: ignore
-
     # testing
     if 'test' in params.settings['cmd']:
         for capability in eval_for: