sklearn.linear_model LogisticRegression选择列

Question

我已从LogisticRegression导入sklearn模型。我的数据集有785列，其中第一列是我试图训练预测的标签。

这是我的代码。

clf = LogisticRegression()
clf = clf.fit(train[[1:785]], train['label'])

我收到以下错误，

File "<ipython-input-22-1c2c333722ad>", line 2
    clf = clf.fit(train[[1:785]], train['label'])
                          ^
SyntaxError: invalid syntax

如何在不明确提及所有列名的情况下选择除第一列之外的所有列。

Answer 1

这对我有用，

clf = clf.fit(train.iloc[:,1:], train[['label']])