对象在试图写入泡菜文件时未找到。

Question

我正在尝试用随机向量森林进行癌症检测。我正在尝试使用命令pickle.dump(forest,open("model.pkl","wb") .But创建一个泡菜文件，我得到了一个名称错误。

NameError                                 Traceback (most recent call last)
c:\Users\hp\newtest\pcancer.ipynb Cell 6' in <cell line: 1>()
----> 1 pickle.dump(forest,open("model.pkl","wb"))

NameError: name 'forest' is not defined

这是我的检测源代码：

import numpy as np
import pandas as pd
import warnings as wr
#Ignoring warnings
from sklearn.exceptions import UndefinedMetricWarning
wr.filterwarnings("ignore", category=UndefinedMetricWarning)
import pickle

df=pd.read_csv('Prostate_cancer_data.csv')
print(df.head(10))
print(df.shape)
print(df.isna().sum())


df=df.dropna(axis=1)#Drop the column with empty data
df=df.drop(['id'],axis=1)


#Encoding first column
from sklearn.preprocessing import LabelEncoder
labelencoder_X=LabelEncoder() 
df.iloc[:,0]=labelencoder_X.fit_transform(df.iloc[:,0].values)

#Splitting data for dependence
X=df.iloc[:,1:].values
Y=df.iloc[:,0].values

#Train-Test split
from sklearn.model_selection import train_test_split
X_train,X_test,Y_train,Y_test=train_test_split(X,Y,test_size=0.25,random_state=1)
 

#Standard scaling
from sklearn.preprocessing import StandardScaler
sc=StandardScaler()
X_train=sc.fit_transform(X_train)
X_test=sc.fit_transform(X_test)



from sklearn.ensemble import RandomForestClassifier
def models(X_train,Y_train):

    #Random forest classifier
    forest=RandomForestClassifier(n_estimators=10,criterion='entropy',random_state=0)
    forest.fit(X_train,Y_train)

    print("Random Forest:",forest.score(X_train,Y_train))

    return forest

    print("Accuracy")
    model=models(X_train,Y_train)

Answer 1

model=models(X_train,Y_train) 

这部分代码不按顺序缩进。因此，它是作为递归调用的本地声明和操作。

Answer 2

代码的最后一节存在缩进问题。这是正确的缩进代码，当您创建一个泡菜文件时，您将在其中写入模型对象，而不是在名为model的对象中返回森林

from sklearn.ensemble import RandomForestClassifier
def models(X_train,Y_train):

    #Random forest classifier
    forest=RandomForestClassifier(n_estimators=10,criterion='entropy',random_state=0)
    forest.fit(X_train,Y_train)

    print("Random Forest:",forest.score(X_train,Y_train))

    return forest

print("Accuracy")
model=models(X_train,Y_train)
pickle.dump(model,open("model.pkl","wb"))