Skip to content

A library that wraps most common usage of scikit-learn for building simple ML models

License

Notifications You must be signed in to change notification settings

SGenheden/ML-playground

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML-playground

Example with Kaggle Titanic data

from ml_playground.data import DataContainer, TextTransformer
from ml_playground.modeller import Modeller
from sklearn.tree import DecisionTreeClassifier

data = DataContainer("../Kaggle/Titanic/train.csv")
data.apply("dropna")
data.transform("Sex", TextTransformer)
data.transform("Embarked", TextTransformer)
data.x = ["Sex", "Pclass", "Age", "SibSp", "Parch", "Embarked"]
data.y = "Survived"

model = Modeller(DecisionTreeClassifier)
model.fit(data)
```

About

A library that wraps most common usage of scikit-learn for building simple ML models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages