Python 多分类任务: 计算每类精度和F1分数并输出表格文件
要获取每类的分类精度并转为F1评价指标,并将结果输出为表格文件,可以按照以下步骤进行修改代码:\n\n1. 导入所需的库和模块,包括pandas和sklearn.metrics中的classification_report和precision_recall_fscore_support。\n\npython\nimport pandas as pd\nfrom sklearn.metrics import classification_report, precision_recall_fscore_support\n\n\n2. 在训练和预测之后,使用classification_report函数计算每类的分类精度。\n\npython\ny_true = # 真实标签\ny_pred = # 预测结果\n\nreport = classification_report(y_true, y_pred)\nprint(report)\n\n\n3. 使用precision_recall_fscore_support函数计算每类的F1评价指标。\n\npython\nprecision, recall, f1, _ = precision_recall_fscore_support(y_true, y_pred)\n\n# 将结果转换为DataFrame\ndf = pd.DataFrame({\n 'Class': range(len(precision)),\n 'Precision': precision,\n 'Recall': recall,\n 'F1-score': f1\n})\n\n\n4. 将结果输出为表格文件,例如CSV文件。\n\npython\ndf.to_csv('classification_metrics.csv', index=False)\n\n\n完整的代码示例如下:\n\npython\nimport pandas as pd\nfrom sklearn.metrics import classification_report, precision_recall_fscore_support\n\n# 训练和预测过程\n\n# 计算每类的分类精度\ny_true = # 真实标签\ny_pred = # 预测结果\n\nreport = classification_report(y_true, y_pred)\nprint(report)\n\n# 计算每类的F1评价指标\nprecision, recall, f1, _ = precision_recall_fscore_support(y_true, y_pred)\n\n# 将结果转换为DataFrame\ndf = pd.DataFrame({\n 'Class': range(len(precision)),\n 'Precision': precision,\n 'Recall': recall,\n 'F1-score': f1\n})\n\n# 将结果输出为表格文件\ndf.to_csv('classification_metrics.csv', index=False)\n\n\n运行代码后,将会在当前目录下生成一个名为classification_metrics.csv的表格文件,其中包含每类的分类精度和F1评价指标。
原文地址: https://www.cveoy.top/t/topic/pP9N 著作权归作者所有。请勿转载和采集!