Skip to content

Draft: Creating predictions for evaluation and plotting

krausef99 requested to merge creating_results_jsons into main

DO NOT MERGE

Adapt predict.py file to enable

  • creating the submission file (predictions for each base for three targets)
  • creating a json with ground truth and predictions as well as attention matrices (averaged over layers and heads)
  • creating a small json as above but with attention for all heads (high dimensionality, thus large storage)
Edited by krausef99

Merge request reports