我正在谷歌云平台ml引擎上的sklearn实现一个简单的k最近邻算法。我使用自定义度量来计算两个输入向量之间的距离,以便距离是两个向量之间的元素平方差中元素的加权和。该代码是下面:真的与这种numpy形状不匹配错误相混淆
import os.path
from sklearn import neighbors
import numpy as np
from six.moves import cPickle as pickle
import tensorflow as tf
from tensorflow.python.lib.io import file_io
flags = tf.app.flags
FLAGS = flags.FLAGS
flags.DEFINE_string('input_dir', 'input', 'Input Directory.')
flags.DEFINE_string('input_train_data','train_data','Input Training Data File Name.')
pickle_file = os.path.join(FLAGS.input_dir, FLAGS.input_train_data)
def mydist(x, y):
return np.dot((x - y) ** 2, weight)
with file_io.FileIO(pickle_file, 'r') as f:
save = pickle.load(f)
train_dataset, train_labels, valid_dataset, valid_labels = save['train_dataset'], save['train_labels'], save[
'valid_dataset'], save['valid_labels']
train_data = train_dataset[:1000]
train_label = train_labels[:1000]
test_data = valid_dataset[:100]
weight = [1.0]* len(train_dataset[1])
knn = neighbors.KNeighborsRegressor(weights='distance', n_neighbors=20, metric=lambda x, y: mydist(x, y))
knn.fit(train_data, train_label)
predict = knn.predict(test_data)
print(predict)
train_dataset是形状(86667,13)和valid_dataset的numpy的阵列具有形状(8000,13)。 Train_labels具有形状(86667,1)和valid_labels(8000,1)。出于某种原因,我得到了一个尺寸不匹配:
line 15, in mydist return np.dot((x - y) ** 2, weight) ValueError: shapes
(10,) and (13,) not aligned: 10 (dim 0) != 13 (dim 0)
X和Y两个自定义指标输入应该有大小13但不知何故,他们有大小10谁能解释一下什么是错在这里?
'重量'的形状是什么?此外,我不熟悉KNeighborRegressor函数,但您在哪里指定x和y是什么? – BenT
weight是一个长度为13的列表。我将自定义度量函数mydist放入KNeighborsRegressor的实例化中的度量参数中。 –