RandomForest IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
I work from sklearn
on RandomForestClassifier
:
class RandomForest(RandomForestClassifier):
def fit(self, x, y):
self.unique_train_y, y_classes = transform_y_vectors_in_classes(y)
return RandomForestClassifier.fit(self, x, y_classes)
def predict(self, x):
y_classes = RandomForestClassifier.predict(self, x)
predictions = transform_classes_in_y_vectors(y_classes, self.unique_train_y)
return predictions
def transform_classes_in_y_vectors(y_classes, unique_train_y):
cyr = [unique_train_y[predicted_index] for predicted_index in y_classes]
predictions = np.array(float(cyr))
return predictions
I got the error:
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
+3
source to share
1 answer
Seems to y_classes
contain values that are not valid indices.
When you try to access unique_train_y
using predicted_index
than you will get an exception as preded_index is not what you think.
Try the following code:
cyr = [unique_train_y[predicted_index] for predicted_index in range(len(y_classes))]
# assuming unique_train_y is a list and predicted_index should be integer.
+1
source to share