Only run predictions

data-processing/granite-ttm.py

@@ -94,20 +94,14 @@ def zeroshot_eval(dataset_name, batch_size, context_length=512, forecast_length=
 
     # train predictions
 
-    print("+" * 20, "Train MSE zero-shot", "+" * 20)
-    zeroshot_output = zeroshot_trainer.evaluate(dset_train)
-    print(zeroshot_output)
-
+    print("+" * 20, "Train predict zero-shot", "+" * 20)
     predictions_dict = zeroshot_trainer.predict(dset_train)
 
     predictions_np_train = predictions_dict.predictions[0]
 
     # test predictions
 
-    print("+" * 20, "Test MSE zero-shot", "+" * 20)
-    zeroshot_output = zeroshot_trainer.evaluate(dset_test)
-    print(zeroshot_output)
-
+    print("+" * 20, "Test predict zero-shot", "+" * 20)
     predictions_dict = zeroshot_trainer.predict(dset_test)
 
     predictions_np_test = predictions_dict.predictions[0]