microsoft / microsoft/FLAML

TypeError: LogisticRegression.fit() got an unexpected keyword argument 'gpu_per_trial'

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

Description

Hello,

I'm experiencing an issue when trying to run AutoFLAML with the `gpu_per_trial` argument. Here is the setup that I'm using:

```python
automl = AutoML()

automl_settings = {
'time_budget': 10,
'metric': 'accuracy',
'task': 'classification',
'estimator_list': 'auto',
'n_jobs': -1,
'use_ray': True,
'n_concurrent_trials': 10,
'gpu_per_trial': 0.1
}

automl.fit( X_train=X, y_train=y, **automl_settings)
```

When I run AutoFLAML with this setup, I get the following error:

`TypeError: LogisticRegression.fit() got an unexpected keyword argument 'gpu_per_trial'`

From my understanding, 'gpu_per_trial' is supposed to be a parameter for the RayTune backend to handle distribution of GPU resources across the trials. However, it seems like this argument is being passed to the fit method of the models AutoFLAML is training, which is causing the error.

I'm currently using AutoFLAML version 1.2.3. Is this a known issue, or is there something I'm missing in my configuration? Any guidance on this would be appreciated.

Thank you!

Best regards,
Andrej

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.