Random forest classifier

Nim implementation of RandomForestClassifier. The interface is a subset of Python's scikit-learn sklearn.ensemble.RandomForestClassifier

Requirements

nim>=1.0

Installation

nimble install randomforest

Usage

import randomforest
clf = RandomForestClassifier(n_estimators=10, criterion="gini")
clf.fit(X_train, y_train)
y = clf.predict(X_test)

Credits

Thank you Jason Brownlee for the article How to Implement Random Forest From Scratch in Python - this implementation was inspired by approach used in the article.

Related projects

DecisionTreeNim - Nim package for decision trees and random forest scikit-learn - Machine learning library for Python - used as reference implementation.

License

GPL-3 Krystian Safjan.