Важность функции случайного леса с использованием Python

Я пытаюсь использовать приведенный ниже код для классификатора случайного леса. Несмотря на то, что я определил, но получаю NameError. Пожалуйста помоги

def RFC_model(randomState, X_train, X_test, y_train, y_test):


   rand_forest = RandomForestClassifier()
   rand_forest.fit(X_train, y_train)
   forest_test_predictions = rand_forest.predict(X_test)
   print(accuracy_score(y_test, forest_test_predictions))

X_train, X_test, y_train, y_test = train_test_split(df_encoded.drop(['success'],axis='columns').values,      
                                                df_encoded.success, 
                                                test_size=0.2)

RFC_model(42, X_train, X_test, y_train, y_test)

0.994045375744328

rand_forest.feature_importances_.round(3)

NameError                                 Traceback (most recent call last)
<ipython-input-40-974786899b7f> in <module>
  1 #importance of features rounded to nearest 3 decimals
----> 2 rand_forest.feature_importances_.round(3)

NameError: name 'rand_forest' is not defined

python random-forest nameerror
person avk    schedule 10.12.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Вы определяете переменную rand_forest локально в области действия функции RFC_model. Как только функция завершает выполнение, объект уничтожается, поэтому вы не можете получить к нему доступ. Вы можете решить эту проблему, вернув объект rand_forest:

def RFC_model(randomState, X_train, X_test, y_train, y_test):
    rand_forest = RandomForestClassifier()
    rand_forest.fit(X_train, y_train)
    forest_test_predictions = rand_forest.predict(X_test)
    print(accuracy_score(y_test, forest_test_predictions))
    return rand_forest

X_train, X_test, y_train, y_test = train_test_split(df_encoded.drop(['success'],axis='columns').values,      
                                            df_encoded.success, 
                                            test_size=0.2)

rand_forest = RFC_model(42, X_train, X_test, y_train, y_test)
rand_forest.feature_importances_.round(3)
person Davide ND    schedule 12.12.2019