-
Notifications
You must be signed in to change notification settings - Fork 7
Expand file tree
/
Copy pathRunCOMET.py
More file actions
15 lines (14 loc) · 811 Bytes
/
Copy pathRunCOMET.py
File metadata and controls
15 lines (14 loc) · 811 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from datasets.Dataset import Etour, Itrust, ItrustFull, Smos, Libest
import comet.CometDataReader as reader
from utility import Util
#result_file = "results_RQ1_iTrust_MAP.tm"
#result_file = "results_RQ1_SMOS_MAP.tm"
#result_file = "results_RQ1_eTour_MAP.tm"
result_file = "results_RQ1_LibEST_MAP.tm"
#dataset = ItrustFull()
#dataset = Etour(use_italian_solution_matrix=True)
#dataset = Smos(use_english_solution_matrix=True)
dataset = Libest()
#reader.eval_f1_comet_data(reader.COMET_DATA_FOLDER / result_file, dataset, Util.get_range_array(0.001, 0.999, 0.001))
#reader.calculate_comet_mean_avg_prec(reader.COMET_DATA_FOLDER / result_file, dataset, None)
reader.convert_comet_to_recall_prec_csv(reader.COMET_DATA_FOLDER / result_file, dataset, Util.get_range_array(0.0, 1.0, 0.001),"libest_comet_curve.csv")