list1
Browse files- src/test_hellaswag.py +1 -1
src/test_hellaswag.py
CHANGED
|
@@ -75,7 +75,7 @@ for idx,batch in enumerate(glue_test_data_loader(input_rng, test_dataset, total_
|
|
| 75 |
restored_output.append(final_output)
|
| 76 |
|
| 77 |
finall=pd.DataFrame({'predictions':restored_output,'permutation':list1})
|
| 78 |
-
finall.to_csv('../predictions.csv')
|
| 79 |
|
| 80 |
|
| 81 |
|
|
|
|
| 75 |
restored_output.append(final_output)
|
| 76 |
|
| 77 |
finall=pd.DataFrame({'predictions':restored_output,'permutation':list1})
|
| 78 |
+
finall.to_csv('../test_predictions/predictions.csv')
|
| 79 |
|
| 80 |
|
| 81 |
|