Skip to main content

Lightwood's goal is to make it very simple for developers to use the power of artificial neural networks in their projects.

Project description


Lightwood Actions workflow PyPI version PyPI - Downloads Discourse posts

Lightwood is like Legos for Machine Learning.

A Pytorch based framework that breaks down machine learning problems into smaller blocks that can be glued together seamlessly with one objective:

  • Make it so simple that you can build predictive models with as little as one line of code.


Learn more from the Lightwood's docs.

Try it out


You can install Lightwood from pip:

pip3 install lightwood

Note: depending on your environment, you might have to use pip instead of pip3 in the above command.


Given the simple sensor_data.csv let's predict sensor3 values. | sensor1 | sensor2 | sensor3 | |----|----|----| | 1 | -1 | -1 | | 0 | 1 | 0 | | -1 |- 1 |1 |

Import Predictor from Lightwood

from lightwood import Predictor

Train the model.

import pandas
sensor3_predictor = Predictor(output=['sensor3']).learn(from_data=pandas.read_csv('sensor_data.csv'))

You can now predict what sensor3 value will be.

prediction = sensor3_predictor.predict(when={'sensor1':1, 'sensor2':-1})
  • You can also try Lightwood in Google Colab: Google Colab


Thanks for your interest. There are many ways to contribute to this project. Please, check out our Contribution guide.

Current contributors

Made with contributors-img.

License PyPI - License

Project details

Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for lightwood, version 0.61.0
Filename, size File type Python version Upload date Hashes
Filename, size lightwood-0.61.0.tar.gz (78.1 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page