【发布时间】:2016-02-10 15:45:00
【问题描述】:
我是 Sklearn 和 python 的新手;我有这个代码 sn-p 用于我试图破译的项目。我希望你们能帮助我。
from repository import Repository
from configuration import config
repository = Repository(config)
dataset, labels = repository.get_dataset_and_labels()
import numpy as np
from sklearn.cross_validation import train_test_split
from sklearn.svm import SVC
from sklearn.cross_validation import ShuffleSplit
from sklearn.grid_search import GridSearchCV
# Ensure that there are no NaNs
dataset = dataset.fillna(-85)
# Split the dataset into training (90 \%) and testing (10 \%)
X_train, X_test, y_train, y_test = train_test_split(dataset, labels, test_size = 0.1 )
cv = ShuffleSplit(X_train.shape[0], n_iter=10, test_size=0.2, random_state=0)
# Define the classifier to use
estimator = SVC(kernel='linear')
# Define parameter space
gammas = np.logspace(-6, -1, 10)
# Use Test dataset and use cross validation to find bet hyper-p rameters.
classifier = GridSearchCV(estimator=estimator, cv=cv, param_grid=dict(gamma=gammas))
classifier.fit(X_train, [repository.locations.keys().index(tuple(l)) for l in y_train])
我无法理解的是分类器的 fit 方法的使用。在我在网上找到的所有示例中,“fit”都会接收训练数据和相应的标签。在上面的示例中,“fit”接收训练数据和标签(而不是标签)的索引。 分类器如何获取索引而不是标签并且仍然有效
【问题讨论】:
-
如果你不介意我问,你从哪里得到的?我问的唯一原因是因为你说它有效。我没有
repository,所以我无法通过查看dataset所指的内容来测试它 -
@Leb 这是我被分配到的一个学术项目的代码 sn-p。数据集由wifi指纹(mac地址,信号级别)组成,标签包含该指纹在建筑物中的坐标。
-
好吧,我的猜测是:
from sklearn.cross_validation import ShuffleSplit返回indices。因此,从外观上看,他们首先做到了这一点,并在GridSearchCV中使用了它。这就是为什么索引被用于y_train的原因,我假设它的形状为 (y,)(即仅列) -
@Leb 所以
classifier.fit实际上是接收一个索引列表,而不是y_train中保存的位置? -
它正在接收一个索引列表,它表示
y_train的位置。这是使用ShuffleSplit进行交叉验证的,这就是为什么您看到cv=cv代表GridSearchCV。换句话说,ShuffleSplit返回索引,为了让这些索引与y_train位置相关,必须传递list,即repository.locations.keys().index(tuple(l))
标签: python machine-learning scikit-learn