yogsototh/clj-ml
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add classifier FastRandomForests

Browse source
This commit is contained in:
Ronny Bjarnason 2011-04-22 15:53:39 -06:00
parent 953cc0bac5
commit 2cb2ccd574
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/clj_ml/classifiers.clj
View file

@ -62,6 +62,7 @@
"
(:use [clj-ml utils data kernel-functions])
(:import (java.util Date Random)
(hr.irb.fastRandomForest FastRandomForest)
(weka.core Instance Instances)
(weka.classifiers.trees J48 RandomForest M5P)
(weka.classifiers.meta LogitBoost AdditiveRegression RotationForest)
@ -191,6 +192,16 @@
:random-seed "-S"
:depth "-depth"}))))
(defmethod make-classifier-options [:decision-tree :fast-random-forest]
([kind algorithm m]
(->>
(check-options m {:debug "-D"})
(check-option-values m
{:num-trees-in-forest "-I"
:num-features-to-consider "-K"
:random-seed "-S"
:depth "-depth"}))))
(defmethod make-classifier-options [:decision-tree :rotation-forest]
([kind algorithm m]
(->>
@ -435,6 +446,10 @@
([kind algorithm & options]
(make-classifier-with kind algorithm RandomForest options)))
(defmethod make-classifier [:decision-tree :fast-random-forest]
([kind algorithm & options]
(make-classifier-with kind algorithm FastRandomForest options)))
(defmethod make-classifier [:decision-tree :rotation-forest]
([kind algorithm & options]
(make-classifier-with kind algorithm RotationForest options)))