RandomForestClassfier.fit (): ValueError: не удалось преобразовать строку в число с плавающей точкой

Данный простой файл CSV:

A,B,C
Hello,Hi,0
Hola,Bueno,1

Очевидно, что настоящий набор данных гораздо сложнее, но этот воспроизводит ошибку. Я пытаюсь построить случайный лесной классификатор для него, например, так:

cols = ['A','B','C']
col_types = {'A': str, 'B': str, 'C': int}
test = pd.read_csv('test.csv', dtype=col_types)

train_y = test['C'] == 1
train_x = test[cols]

clf_rf = RandomForestClassifier(n_estimators=50)
clf_rf.fit(train_x, train_y)

Но я просто получаю эту трассировку при вызове fit ():

ValueError: could not convert string to float: 'Bueno'

версия scikit-learn - 0.16.1.

Ответы на вопрос(5)

Ваш ответ на вопрос