cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results for 
Search instead for 
Did you mean: 

Customizing class imbalance handling in databricks.automl

rtreves
New Contributor III

Hi Databricks! I'd like to make a feature request for end users to be able to customize class imbalance behavior in `databricks.automl`, specifically `databricks.automl.classify`. This public documentation describes the default procedure, but my team is unaware of a way for users to modify this behavior. Specifically, we'd like to be able to:

  • Toggle class imbalance handling on/off
  • Change the threshold at which a dataset is considered imbalanced
  • Change the sampling fraction used in downsampling the major class

In the long term, availability of more complex imbalance correction techniques such as SMOTE would be appreciated too.

Thank you.

0 REPLIES 0

Connect with Databricks Users in Your Area

Join a Regional User Group to connect with local Databricks users. Events will be happening in your city, and you won’t want to miss the chance to attend and share knowledge.

If there isn’t a group near you, start one and help create a community that brings people together.

Request a New Group