Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 633 Bytes

README.md

File metadata and controls

32 lines (23 loc) · 633 Bytes

koala

koala allows you to rapidly train a random forest classifier. Useful for quick prototyping.

####Requirements

  • Python 3.3+
  • NumPy
  • scikit-learn
  • pandas
  • copper

####Installation $ python setup.py install

####Basics

from koala import Koala
import pandas as pd

wine = pd.read_csv('../data/wine.csv')
kl = Koala(data=wine, target='Type')
kl.train(test_size=0.3)
print(kl.confusion_matrix())

    1   2   3
1  23   0   0
2   1  17   0
3   0   0  13

More information and code samples available in the notebooks folder.