Я использую cuML для стохастического градиентного спуска. Я использовал sklearn's train_test_split для генерации разбиений для train_X, train_y ... из кадра данных cuDF.
Следующий код (я удалил гиперпараметры, не относящиеся к этому вопросу):
from cuml.solvers import SGD as cumlSGD
cu_sgd = cumlSGD(eta0=0.005)
cu_sgd.fit(train_X, train_y)
Выдает следующую ошибку в строке cu_sgd.fit:
У объекта 'nvstrings' нет атрибута 'to_gpu_array'
Как мне обойти эту проблему?