Lightwood:基于Pytorch的机器学习模块化框架
一个基于Pytorch的框架,将机器学习问题分解成更小的块,可以用一个目标无缝粘合在一起。
Python 机器学习
共84Star
详细介绍
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.
Documentation
Learn more from the Lightwood's docs.
Try it out
Installation
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.
Usage
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})
Contributing
Thanks for your interest.There are many ways to contribute to this project. Get started here.
License
-
0 Star
-
816 Star
-
215 Star
-
604 Star
-
222 Star
-
856 Star
-
865 Star
-
1939 Star