8000 GitHub - frederikhoengaard/lazy-learn at dc9ecdb9620b64e121a23c1e23147f352d22fb83
[go: up one dir, main page]

Skip to content

frederikhoengaard/lazy-learn

Repository files navigation

lazy-learn is a high-level Python interface for automated machine learning (AutoML). While there are many AutoML libraries available each typically solves a niche area of the overall ML pipeline without providing a covering and approachable end-to-end system.

The aim of lazy-learn is exactly that. Given a dataset, lazy-learn will analyse types and distributions of attributes, preprocess, feature-engineer and ultimately train models to be used for further evaluation or inference.

Upcoming features

Current stable version is 0.0.3. The upcoming updates will support:

  • Abstract construction of model architectures
  • XGBoost, LightGBM, Adaboost and Catboost architectures
  • Time partitioning of datasets
  • Automated Hyperparameter Optimisation (HPO)
  • Text features
  • An interface to AutoGluon
  • Outlier detection and handling
  • Automated suggestions of performance metrics

Usage

Using lazy-learn revolves around the LazyLearner class. You can think of it as a kind of project, and it is the wrapper for any experiment within lazy-learn.

Installation

Dependencies

lazy-learn requires:

  • pandas
  • scikit-learn
  • xgboost

User Installation

pip install lazy-learn

Help and Support

Documentation

Citation

0