microsoft / FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
https://microsoft.github.io/FLAML/
MIT License
3.75k stars 495 forks source link

features: Manually setting the validation set for multi-output task #1302

Open lizhuoq opened 1 month ago

lizhuoq commented 1 month ago

Why are these changes needed?

For original multi-output tasks where the eval_method is holdout, manual setting of the validation set was not possible. This commit introduces a new feature allowing manual setting of the validation set for multi-output tasks.

model = MultiOutputRegressor(
    AutoML(
        task="regression",
        time_budget=1,
        eval_method="holdout",
        multioutput_train_size=len(X_train)
    )
)
model.fit(
    pd.concat([X_train, X_val]),
    pd.concat([y_train, y_val])
)

Related issue number

Checks

lizhuoq commented 1 month ago

@microsoft-github-policy-service agree