Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/bistcuite/plainml

Painless Machine Learning Library for python based on scikit-learn
https://github.com/bistcuite/plainml

machine-learning ml plainml python scikit-learn

Last synced: 18 days ago
JSON representation

Painless Machine Learning Library for python based on scikit-learn

Awesome Lists containing this project

README

        

# PlainML
Painless Machine Learning Library for python based on [scikit-learn](https://scikit-learn.org/stable/).

[![Pypi](https://img.shields.io/pypi/v/plainml)](https://pypi.org/project/plainml)
[![Downloads](https://pepy.tech/badge/plainml)](https://pepy.tech/project/plainml)

## Installation
```
pip install plainml
```

## Example
```py
from plainml import KnnModel, load_iris, train_test_split

dt = load_iris()
data = dt.data
target = dt.target

x_train, x_test, y_train, y_test = train_test_split(data, target, test_size=0.2)

model = KnnModel(x_train, y_train)
model.fit()

print(model.score(x_test, y_test))
print(model.predict([[5.1, 3.5, 1.4, 0.2]]))

model.save(file_name='iris_knn.pkl')
```

## Todo
- `run_api`**WIP**
- `create_dashboard`
- AutoML

## License
[MIT License](LICENSE)