Изменения

Перейти к: навигация, поиск

Дерево решений и случайный лес

1656 байт добавлено, 01:57, 29 ноября 2020
Пример на языке R
Пример классификации с применением <code>weka.classifiers.trees.RandomForest</code><ref>[http://weka.sourceforge.net/doc.dev/weka/classifiers/trees/RandomForest.html Weka, Random Forest]</ref>
<code>Maven </code> зависимость:
<dependency>
<groupId>nz.ac.waikato.cms.weka</groupId>
'''import''' weka.classifiers.trees.RandomForest;
<font color="green">// read dataset</font>
'''var''' trainingDataSet = getDataSet(...);
'''var''' testingDataSet = getDataSet(...);
<font color="green">// create random forest classifier</font>
'''var''' forest = new RandomForest();
forest.setMaxDepth(15);
forest.setNumFeatures(2);
forest.buildClassifier(trainingDataSet);
<font color="green">// evaluate the model on test dataset and print summary</font>
'''var''' eval = new Evaluation(trainingDataSet);
eval.evaluateModel(forest, testingDataSet);
System.out.println(eval.toSummaryString());
=== Пример на языке R ===
{{Main|Примеры кода на R}}
==== Деревья решений ====
Для создания деревьев решений используется функция <code>ctree()</code> из пакета <code>party</code>.
<font color="gray"># importing package </font>
install.packages(<font color="green">"party"</font>)
<font color="gray"># reading data</font>
rdata <- read.csv(<font color="green">"input.csv"</font>, <font color="#660099">sep</font> = <font color="green">','</font>, <font color="#660099">header</font> = FALSE)
<font color="gray"># evaluating model</font>
output.tree <- ctree(target ~ x + y + z, <font color="#660099">data</font> = rdata)
<font color="gray"># plotting results</font>
plot(output.tree)
 
==== Случайный лес ====
Для создания случайного леса необходимо импортировать пакет <code>randomForest</code>
 
<font color="gray"># importing packages </font>
install.packages(<font color="green">"party"</font>)
install.packages(<font color="green">"randomForest"</font>)
<font color="gray"># reading data</font>
rdata <- read.csv(<font color="green">"input.csv"</font>, <font color="#660099">sep</font> = <font color="green">','</font>, <font color="#660099">header</font> = FALSE)
<font color="gray"># creating the forest</font>
output.forest <- randomForest(target ~ x + y + z, <font color="#660099">data</font> = rdata)
<font color="gray"># getting results</font>
print(output.forest)
== См. также ==
286
правок

Навигация