1

我正在尝试比较不同的算法，看看哪种算法最适合我的问题。

我正在直接从本教程中试用代码：https ://machinelearningmastery.com/machine-learning-in-python-step-by-step/

特别是在下面的代码中：

我的进口

import sys
import pandas as pd
import scipy as sp
import sklearn as sk
import numpy as np
import matplotlib.pyplot as plt
from pandas.plotting import scatter_matrix

from sklearn.model_selection import train_test_split
from sklearn.model_selection import KFold

from sklearn.linear_model import LogisticRegression
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
from sklearn.neighbors import KNeighborsClassifier
from sklearn.svm import SVC
from sklearn.naive_bayes import GaussianNB
from sklearn.tree import DecisionTreeClassifier

抽查算法

models = []
models.append(('LR', LogisticRegression(solver='liblinear', multi_class='ovr')))
models.append(('LDA', LinearDiscriminantAnalysis()))
models.append(('KNN', KNeighborsClassifier()))
models.append(('CART', DecisionTreeClassifier()))
models.append(('NB', GaussianNB()))
models.append(('SVM', SVC(gamma='auto')))

依次评估每个模型

results = []
names = []
for name, model in models:
    kfold = model_selection.KFold(n_splits=10, random_state=seed)
    cv_results = model_selection.cross_val_score(model, X_train, Y_train, cv=kfold, scoring=scoring)
    results.append(cv_results)
    names.append(name)
    msg = "%s: %f (%f)" % (name, cv_results.mean(), cv_results.std())
    print(msg)

当我运行它时，我不断得到（<---- 第 12 行）：

NameError Traceback (most recent call last)
<ipython-input-25-e6a861b6e218> in <module>()
     10 names = []

     11 for name, model in models:

     12         kfold = model_selection.KFold(n_splits=10, random_state=seed) <----

     13         cv_results = model_selection.cross_val_score(model, X_train, Y_train, cv=kfold, scoring=scoring)

     14         results.append(cv_results)

NameError: name 'model_selection' is not defined

有人可以向我解释 KFold 是如何工作的以及为什么它不接受该实例吗？

4

1 回答 1

5

KFold是sklearn.model_selection模块的一部分。

确保将名称导入您的工作区，或者通过执行

from sklearn import model_selection

并使用

model_selection.KFold

或者

import sklearn.model_selection
sklearn.model_selection.KFold

甚至

from sklearn.model_selection import KFold
KFold
于 2019-04-21T16:46:42.707 回答