使用 sklearn 的 'predict' 函数

Question

如果我在 sklearn 中训练了一个模型，使用虚拟变量作为分类值，那么将单行特征输入该模型以获得预测结果的最佳做法是什么？对于所有输入数据集，我都试图获得分数。如果我的列少于我用于 train/fit 模型的数据集，它会抛出错误吗？

澄清一下：在构建模型之前，我采用了一个包含 5 列的数据集并创建了超过 118 个虚拟列。现在我有一行包含 5 列的数据，我想在 predict 函数中使用它们。我该怎么做？

如有任何帮助，我们将不胜感激。

Answer 1

根据 table 状态扩展特征是错误的，因为您不能用其他数据重复它。如果你想以这种方式创建特征，你应该使用一个会记住特征结构的构造函数。由于您没有提供数据示例，因此这里是如何制作构造函数的主要思想：

import pandas as pd

data = pd.DataFrame([['Missouri', 'center', 'Jan', 55, 11],
                     ['Kansas', 'center', 'Mar', 54, 31],
                     ['Georgia', 'east', 'Jan', 37, 18]],
                     columns=('state', 'pos', 'month', 'High Temp', 'Low Temp'))


test =  pd.DataFrame([['Missouri', 'center', 'Feb', 44, 23], 
                      ['Missouri', 'center', 'Mar', 55, 33]],
                      columns=('state', 'pos', 'month', 'High Temp', 'Low Temp'))  


class DummyColumns():
    def __init__(self, data):
        # Columns constructor
        self.empty = pd.DataFrame(columns=(list(data.columns) +
                                           list(data.state.unique()) +
                                           list(data.pos.unique()) +
                                           ['Winter', 'Not winter']))
    def __call__(self, data):
        # Initializing with zeros
        self.df = pd.DataFrame(data=0, columns=self.empty.columns, index=data.index)        
        for row in data.itertuples():
            self.df.loc[row.Index, :5] = row[1:]
            self.df.loc[row.Index, row.state] = 1
            self.df.loc[row.Index, row.pos] = 1
            if row.month in ['Dec', 'Jan', 'Feb']:
                self.df.loc[row.Index, 'Winter'] = 1
            else:
                self.df.loc[row.Index, 'Not winter'] = 1
        return self.df       

add_dummy = DummyColumns(data)
dummy_test = add_dummy(test)
print dummy_test

      state     pos month  High Temp  Low Temp  Missouri  Kansas  Georgia  \
0  Missouri  center   Feb         44        23         1       0        0   
1  Missouri  center   Mar         55        33         1       0        0   

   center  east  Winter  Not winter  
0       1     0       1           0  
1       1     0       0           1

使用 sklearn 的 'predict' 函数

Using sklearn's 'predict' function

python

svm

predict

random-forest

scikit-learn