python - 为校准模型拆分保留数据

Question

我需要拆分我的进一步数据集以拟合校准模型..但我不知道如何进一步拆分它。见下文

我已经用下面的方法训练了我的模型

Train =  pd.read_csv("Test.csv")
ytrain = Train['Fraud']
Xtrain = Train.drop( ['Fraud'], axis = 1 )

我如何进一步拆分以下数据以适合我的校准模型

Test = pd.read_csv("Test.csv")
ytest = Test['Fraud']
Xtest = Test.drop( ['Fraud'], axis = 1 )

标签： pythonpandasmachine-learningscikit-learn

Answer 1

如果您使用CalibratedClassifierCV()，它将根据用于交叉验证的测试样本进行校准。如果您想手动选择用于校准的数据，您可以设置参数cv="prefit"，然后设置新.fit的CalibratedClassifierCV验证集。