1. ホーム
  2. スクリプト・コラム
  3. パイソン

[解決済み】IndexError: invalid index to scalar variableを修正する方法

2022-01-12 05:31:15

質問

私のコードは以下の通りです。



import pandas as pd
import numpy as np
from sklearn import ensemble
from sklearn import cross_validation

def ToWeight(y):
    w = np.zeros(y.shape, dtype=float)
    ind = y != 0
    w[ind] = 1./(y[ind]**2)
    return w

def RMSPE(y, yhat):
    w = ToWeight(y)
    rmspe = np.sqrt(np.mean( w * (y - yhat)**2 ))
    return rmspe

forest = ensemble.RandomForestRegressor(n_estimators=10, min_samples_split=2, n_jobs=-1)

print ("Cross validations")
cv = cross_validation.KFold(len(train), n_folds=5)

results = []
for traincv, testcv in cv:
    y_test = np.expm1(forest.fit(X_train[traincv], y_train[traincv]).predict(X_train[testcv]))
    results.append(RMSPE(np.expm1(y_train[testcv]), [y[1] for y in y_test]))

testcv があります。

[False False False ...,  True  True  True]

実行すると、エラーが発生します。

IndexError: invalid index to scalar variable.

を行に追加してください。 results.append(RMSPE(np.expm1(y_train[testcv]), [y[1] for y in y_test]))

解決方法は?

スカラー値(反復不可能な値)にインデックスを作成しようとしています。

[y[1] for y in y_test]
#  ^ this is the problem

を呼び出すと [y for y in test] ではすでに値に対して繰り返し処理を行なっているので y .

あなたのコードは、次のようにしようとしているのと同じです。

y_test = [1, 2, 3]
y = y_test[0] # y = 1
print(y[0]) # this line will fail

結果の配列に何を入れようとしているのかよくわかりませんが、とりあえずは [y[1] for y in y_test] .

y_testの各yをresultsに追加したい場合は、リスト内包をさらに拡張して以下のようにします。

[results.append(..., y) for y in y_test]

あるいは、forループを使うだけです。

for y in y_test:
    results.append(..., y)