У меня есть следующий DataFrame:
features_train = train_set[["budget", "production_companies_one_hot_enc"]]
label_train = train_set[["popularity"]]
features_train.shape # (3842, 2)
label_train.shape # (3842, 1)
lin_reg = LinearRegression()
lin_reg.fit(features_train, label_train)
Код выдает ValueError: setting an array element with a sequence.
features_train.head()
budget production_companies_one_hot_enc
4687 0 [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, ...
2951 11000000 [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, ...
4071 2000000 [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, ...
4579 400000 [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, ...
2197 20000000 [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, ...
, поэтому столбец production_companies_one_hot_enc содержит один массив горячего кодирования. Эта входная форма действительна?