Commit 9e852d7b by Pamela Osuna

corrections

parent aa5b878c
...@@ -16,7 +16,7 @@ def create_pr(num_classes, y_test, y_pred): ...@@ -16,7 +16,7 @@ def create_pr(num_classes, y_test, y_pred):
average_precision["micro"] = average_precision_score(y_test, y_pred, average="micro") average_precision["micro"] = average_precision_score(y_test, y_pred, average="micro")
print('Average precision score, micro-averaged over all classes: {0:0.2f}'.format(average_precision["micro"])) print('Average precision score, micro-averaged over all classes: {0:0.2f}'.format(average_precision["micro"]))
return recall, precision, avg_precision return recall, precision, average_precision
def plot_pr(recall, precision, average_precision): def plot_pr(recall, precision, average_precision):
#plotting #plotting
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment