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


License
MIT
Install
pip install lightwood==0.11.3

Documentation

Lightwood

Build Status PyPI version

Lightwood has 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 was inspired on Ludwig but runs on Pytorch and gives you full control of what you can do.

Documentation

Learn more Lightwood's docs

Quick start

pip3 install lightwood

Learn

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'))

Predict

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

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