Skip to content

bistcuite/plainml

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PlainML

Painless Machine Learning Library for python based on scikit-learn.

Pypi Downloads

Installation

pip install plainml

Example

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_apiWIP
  • create_dashboard
  • AutoML

License

MIT License