Error while fitting

[GDGauravDutta]

AssertionError Traceback (most recent call last) /tmp/ipykernel_33/4176855811.py in ----> 1 automl.fit(X_train, y_train, task="regression",metric='rmse',time_budget=3600)

/opt/conda/lib/python3.7/site-packages/flaml/automl.py in fit(self, X_train, y_train, dataframe, label, metric, task, n_jobs, log_file_name, estimator_list, time_budget, max_iter, sample, ensemble, eval_method, log_type, model_history, split_ratio, n_splits, log_training_metric, mem_thres, pred_time_limit, train_time_limit, X_val, y_val, sample_weight_val, groups_val, groups, verbose, retrain_full, split_type, learner_selector, hpo_method, starting_points, seed, n_concurrent_trials, keep_search_state, early_stop, append_log, auto_augment, min_sample_size, use_ray, metric_constraints, **fit_kwargs) 2089 2090 self._validate_data( -> 2091 X_train, y_train, dataframe, label, X_val, y_val, groups_val, groups 2092 ) 2093 self._search_states = {} # key: estimator name; value: SearchState

/opt/conda/lib/python3.7/site-packages/flaml/automl.py in _validate_data(self, X_train_all, y_train_all, dataframe, label, X_val, y_val, groups_val, groups) 887 assert isinstance(y_train_all, np.ndarray) or isinstance( 888 y_train_all, pd.Series --> 889 ), "y_train_all must be a numpy array or a pandas series." 890 assert ( 891 X_train_all.size != 0 and y_train_all.size != 0

AssertionError: y_train_all must be a numpy array or a pandas series.

Info about data

<class 'pandas.core.frame.DataFrame'> RangeIndex: 15129 entries, 0 to 15128 Data columns (total 28 columns):

Column Non-Null Count Dtype

---

0 bedrooms 15129 non-null uint8
1 bathrooms 15129 non-null float32 2 sqft_living 15129 non-null uint16 3 sqft_lot 15129 non-null uint32 4 floors 15129 non-null float32 5 waterfront 15129 non-null uint8
6 view 15129 non-null uint8
7 condition 15129 non-null uint8
8 grade 15129 non-null uint8
9 sqft_above 15129 non-null uint16 10 sqft_basement 15129 non-null uint16 11 yr_built 15129 non-null uint16 12 yr_renovated 15129 non-null uint16 13 lat 15129 non-null float32 14 long 15129 non-null float32 15 dateyear 15129 non-null uint16 16 datequarter 15129 non-null uint8
17 datemonth 15129 non-null uint8
18 dateday 15129 non-null uint8
19 dateday_of_week 15129 non-null uint8
20 dateday_of_year 15129 non-null uint16 21 dateweekofyear 15129 non-null uint8
22 dateis_month_end 15129 non-null uint8
23 dateis_month_start 15129 non-null uint8
24 dateis_quarter_end 15129 non-null uint8
25 dateis_quarter_start 15129 non-null uint8
26 dateis_year_end 15129 non-null uint8
27 dateis_weekend 15129 non-null uint8
dtypes: float32(4), uint16(7), uint32(1), uint8(16)

[qingyun-wu]

Hi @GDGauravDutta, is your y_train_all a pandas series? Make sure it is in pandas series format. Let me know if it does not resolve the issue. Thank you!

[GDGauravDutta]

Hi @GDGauravDutta, is your y_train_all a pandas series? Make sure it is in pandas series format. Let me know if it does not resolve the issue. Thank you!

All are series

[qingyun-wu]

Hi @GDGauravDutta, Can you share with me a copy or a sub-sample of your data such that I can reproduce the error you had? Thank you!