是否可以将ROC曲线记录到CSV文件中

Question

我在python上使用sklearn生成了一条ROC曲线，想知道是否有办法将结果记录到CSV文件中？

我能够使用keras 的以下CSV记录器将模型训练数据记录到CSV

csv_logger = CSVLogger('training.log')
model.fit(X_train, Y_train, callbacks=[csv_logger])

如果ROC曲线有类似的东西，那将是完美的

Answer 1

您想将其作为回调还是后期拟合？

帖子：

import pandas as pd
from sklearn import metrics
...
fpr, tpr, thresholds = metrics.roc_curve(...)
pd.DataFrame([fpr, tpr, thresholds]).to_csv("ROC.csv")
...
roc = pd.read_csv("ROC.csv")