Deep Learning with PyTorch made easy 🚀 !

Carefree?#

carefree-learn aims to provide CAREFREE usages for both users and developers. It also provides a corresponding repo for production.

Machine Learning 📈#

Computer Vision 🖼️#

Why carefree-learn?#

carefree-learn is a general Deep Learning framework based on PyTorch. Since v0.2.x, carefree-learn has extended its usage from tabular dataset to (almost) all kinds of dataset. In the mean time, the APIs remain (almost) the same as v0.1.x: still simple, powerful and easy to use!

Here are some main advantages that carefree-learn holds:

Machine Learning 📈#

• Provides a scikit-learn-like interface with much more 'carefree' usages, including:
  • Automatically deals with data pre-processing.
  • Automatically handles datasets saved in files (.txt, .csv).
  • Supports Distributed Training, which means hyper-parameter tuning can be very efficient in carefree-learn.
• Includes some brand new techniques which may boost vanilla Neural Network (NN) performances on tabular datasets, including:
  • TreeDNN with Dynamic Soft Pruning, which makes NN less sensitive to hyper-parameters.
  • Deep Distribution Regression (DDR), which is capable of modeling the entire conditional distribution with one single NN model.
• Supports many convenient functionality in deep learning, including:
  • Early stopping.
  • Model persistence.
  • Learning rate schedulers.
  • And more...
• Full utilization of the WIP ecosystem cf*, such as:
  • carefree-toolkit: provides a lot of utility classes & functions which are 'stand alone' and can be leveraged in your own projects.
  • carefree-data: a lightweight tool to read -> convert -> process ANY tabular datasets. It also utilizes cython to accelerate critical procedures.

From the above, it comes out that carefree-learn could be treated as a minimal Automatic Machine Learning (AutoML) solution for tabular datasets when it is fully utilized. However, this is not built on the sacrifice of flexibility. In fact, the functionality we've mentioned are all wrapped into individual modules in carefree-learn and allow users to customize them easily.

Computer Vision 🖼️#

• Also provides a scikit-learn-like interface with much more 'carefree' usages.
• Provides many out-of-the-box pre-trained models and well hand-crafted training defaults for reproduction & finetuning.
• Seamlessly supports efficient ddp (simply call cflearn.api.run_ddp("run.py"), where run.py is your normal training script).
• Bunch of utility functions for research and production.

Citation#

If you use carefree-learn in your research, we would greatly appreciate if you cite this library using this Bibtex:

License#

carefree-learn is MIT licensed, as found in the LICENSE file.