microsoft / microsoft/FLAML

Support y as a single-column pd.DataFrame

Open
#638 3 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Jupyter Notebook
Stars
4.4k
Forks
565
Avg merge
5d 8m
Merged PRs (30d)
17

Description

When using` FLAML(task="classification")` as a component model in EconML/DoWhy, sometimes it gets called with `y` being a `pd.DataFrame` with a single column, and that throws an error as it only expects a `pd.Series` or `np.ndarray`.

Can we please also support dataframes with a single column, in which case we'd take that column as` y`?

Currently I monkey-patch around it, but it would be nicer if it just handled it.

Happy to do the PR myself if you tell me where in the code the changes should live :)

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.