Список 2D фигур в Python, используемых в качестве параметра в несбалансированном обучении SMOTE? - PullRequest
1 голос
/ 03 июля 2019

Я пытаюсь выполнить передискретизацию своего набора данных в проблеме множественной регрессии.Для передискретизации я использую библиотеку imblearn.over_sampling и метод SMOTE.

Возможны форма оружия (размер 16) и форма карты (размер 20,20,7).Цели имеют форму (размер 16).

Я выполняю разделение теста поезда, где извлекаются train_x и train_y

train_x = [weapons[:train_size, :], maps[:train_size, :]]
train_y = targets[:train_size]

Затем с помощью SMOTE:

sm = SMOTE(random_state=42)
train_x_oversampled, train_y_oversampled = sm.fit_sample(train_x, train_y)

Но я получаю ошибку "ValueError: не удалось передать входной массив из фигуры (280000,16) в фигуру (280000)"

Я попытался преобразовать список в ndarrayиспользуя np.asarray, но проблема все еще остается.

Есть какие-нибудь мысли по этому поводу?

...