基于GBDT回归模型的情况#1.读取数据importnumpyasnpimportpandasaspddf=PD.read _ excel (信用记分卡模型. xlsx ' )打印) df.head 3 .将训练集与测试集分开的froms klearn.model _ selectionimporttrain _ test _ splitx _ train,X_test, y_train random_state=123 ) #4.模型培训和froms klearn.ensembleimportgradientboostingregressormodel=gradientboostion 5 .模型预测和评估y_pred=model.predict ) x_test ) #print ) y_pred[0:10] ) y_pred_new=NP.Rooin 2)有两个数量级
转换为两位有效数字的要点如下
在y_pred_new=NP.round(y_pred,2 )之前尝试了
print (round (y _ pred [ 0:10,2 ) ) ) )有点异想天开,单个数字可以,但在多维数组中会出现以下错误:
typeerror : type numpy.ndarraydoesn ' t define _ round _ _ method
重要的是,将列表转换为数组时,必须采用round(y_pred,2 )中有效的两位数字并将数组转换为列表。