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 from the Lightwood's docs.
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})