Refactor ISCO hierarchy creation and update metric calculation***
Browse files- metric_template_1.py +4 -7
metric_template_1.py
CHANGED
|
@@ -94,10 +94,8 @@ class MetricTemplate1(evaluate.Metric):
|
|
| 94 |
|
| 95 |
# Download and prepare the ISCO structure csv file
|
| 96 |
isco_csv = dl_manager.download_and_extract(ISCO_CSV_URL)
|
| 97 |
-
isco_json = dl_manager.download_and_extract(ISCO_JSON_URL)
|
| 98 |
print(f"ISCO CSV file downloaded")
|
| 99 |
-
|
| 100 |
-
self.isco_hierarchy = isco.create_hierarchy_tree(isco_json)
|
| 101 |
print("ISCO hierarchy dictionary created")
|
| 102 |
print(self.isco_hierarchy)
|
| 103 |
|
|
@@ -119,11 +117,10 @@ class MetricTemplate1(evaluate.Metric):
|
|
| 119 |
# true_labels = [{'G'}]
|
| 120 |
# predicted_labels = [{'F'}]
|
| 121 |
hierarchy = self.isco_hierarchy
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
references, predictions, tree, code_to_node
|
| 125 |
)
|
| 126 |
-
|
| 127 |
print(
|
| 128 |
f"Hierarchical Precision: {hP}, Hierarchical Recall: {hR}, Hierarchical F-measure: {hF}"
|
| 129 |
)
|
|
|
|
| 94 |
|
| 95 |
# Download and prepare the ISCO structure csv file
|
| 96 |
isco_csv = dl_manager.download_and_extract(ISCO_CSV_URL)
|
|
|
|
| 97 |
print(f"ISCO CSV file downloaded")
|
| 98 |
+
self.isco_hierarchy = isco.create_hierarchy_dict(isco_csv)
|
|
|
|
| 99 |
print("ISCO hierarchy dictionary created")
|
| 100 |
print(self.isco_hierarchy)
|
| 101 |
|
|
|
|
| 117 |
# true_labels = [{'G'}]
|
| 118 |
# predicted_labels = [{'F'}]
|
| 119 |
hierarchy = self.isco_hierarchy
|
| 120 |
+
hP, hR = ham.calculate_hierarchical_precision_recall(
|
| 121 |
+
references, predictions, hierarchy
|
|
|
|
| 122 |
)
|
| 123 |
+
hF = ham.hierarchical_f_measure(hP, hR)
|
| 124 |
print(
|
| 125 |
f"Hierarchical Precision: {hP}, Hierarchical Recall: {hR}, Hierarchical F-measure: {hF}"
|
| 126 |
)
|