def _compile_table(self, name):
# for readability
metrics = self.metrics[name]
fn = metrics['fn']; fp = metrics['fp']; tp = metrics['tp']; tn = metrics['tn']
pt = PrettyTable([Fore.GREEN + name + Style.RESET_ALL,'Predicted NO','Predicted YES','Total'])
pt.add_row(['Actual NO',tn,fp,tn+fp])
pt.add_row(['Actual YES',fn,tp,fn+tp])
pt.add_row(['Total',tn+fn,fp+tp,''])
pt.hrules = ALL
rows = ['' for i in range(6)]
rows[0] = pt.get_string(padding_width=5)
rows[1] = "Accuracy: {:^1}{:<.2f}".format("", metrics['accuracy'])
rows[2] = "Precision: {:^}{:<.2f}".format("", metrics['precision'])
rows[3] = "Recall: {:^3}{:<.2f}".format("", metrics['recall'])
rows[4] = "F1-score: {:^1}{:<.2f}".format("", metrics['f1_score'])
rows[5] = ""
self.rows.extend(rows)
评论列表
文章目录