Я начинаю осваивать машинное обучение в scikit-learn.
Я написала 2 функции, одна обучает, другая тестирует.
def log_reg_train(self, data_train, target_train): logreg = linear_model.LogisticRegression() logreg.fit(data_train, target_train) #обучение self.__st_model = pickle.dumps(logreg) #сохранение модели def log_reg_test(self, data_test, target_test): data_test_norm = self.normalization(data_test) #нормализация тестовых данных logreg = pickle.loads(self.__st_model) #загрузка сохранённой модели predicted = logreg.predict(data_test_norm) #тестирование print(classification_report(target_test, predicted)) #вывод результатов
Вылазит ошибка:
... in check_array array = array.astype(np.float64) ValueError: setting an array element with a sequence.
Не могу понять, что это значит и в чём именно дело. Может я вообще что-то не так делаю?
Подскажите, пожалуйста!