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


Build Status PyPI version

Lightwood is like Legos for Machine Learning, with two objectives:

  • Make it so simple that you can build predictive models with a line of code.
  • Make it so flexible that you can change and customize everything.

Lightwood runs on Pytorch and gives you full control of what you can do.


Learn more Lightwood's docs

Quick start

pip3 install lightwood


You can train a Predictor as follows:

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


You can now given new readings from sensor1 and sensor2 predict what sensor3 will be.

prediction = sensor3_predictor.predict(when={'sensor1':1, 'sensor2':-1})

Project details

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.11.7
Filename, size File type Python version Upload date Hashes
Filename, size lightwood-0.11.7.tar.gz (49.4 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page