sklearn Gradient Boosting Classifier の estimator.loss_ メソッドに問題があります。時間の経過に伴うトレーニング エラーと比較して、テスト エラーをグラフ化しようとしています。これが私のデータ準備の一部です:
# convert data to numpy array
train = np.array(shuffled_ds)
#label encode neighborhoods
for i in range(train.shape[1]):
if i in [1,2]:
print(i,list(train[1:5,i]))
lbl = preprocessing.LabelEncoder()
lbl.fit(list(train[:,i]))
train[:,i] = lbl.transform(train[:,i])
print('neighborhoods & crimes encoded')
#create target vector
y_crimes = train[::,1]
train=np.delete(train,1,1)
print(y_crimes)
#arrays to float
train = train.astype(float)
y_crimes = y_crimes.astype(float)
#data holdout for testing
X_train, X_test, y_train, y_test = cross_validation.train_test_split(
train, y_crimes, test_size=0.4, random_state=0)
print('test data created')
#train model and check train vs test error
print('begin training...')
est=GBC(n_estimators = 3000,learning_rate=.1,max_depth=4,max_features=1,min_samples_leaf=3)
est.fit(X_train,y_train)
print('done training')
この時点で、配列の形状を
print(X_train.shape)
print(X_test.shape)
print(y_train.shape)
print(y_test.shape)
私は得る:
(18000, 9)
(12000, 9)
(18000,)
(12000,)
それぞれ。
したがって、sklearnのドキュメントによると、私の形状は互換性があります。しかし次に、テスト スコア ベクトルを入力して、トレーニング エラーと比較するためにグラフ化します。
test_score=np.empty(len(est.estimators_))
for i, pred in enumerate(est.staged_predict(X_test)):
test_score[i] = est.loss_(y_test,pred)
次のエラーが表示されます。
: operands could not be broadcast together with shapes (12000,47) (12000,)
return np.sum(-1 * (Y * pred).sum(axis=1) +
543 544else:ValueError
その47がどこから来ているのかわかりません。以前に別のデータセットで同じ手順を使用しましたが、問題はありませんでした。どんな助けでも大歓迎です。