init: Model initialization

In pievos101/DFNET: Network-guided greedy decision forest for feature subset selection

Model initialization

Description

Initializes the decision forest network.

Usage

init(
  graph,
  features,
  target,
  ntrees = 100,
  walk.depth = NaN,
  performance = NULL,
  flatten.sep = "$",
  importance = "impurity",
  splitrule = "gini"
)

Arguments

graph	The graph to train the network on.
features	numeric matrix or 3D array. The features to train on.
target	numeric vector. The target to train towards.
ntrees	integer. The number of trees to generate per iteration.
walk.depth	integer. The number of nodes to select per module.
performance	unary function. Called with a decision tree as argument to estimate that tree's performance.
flatten.sep	string. Separator to use when flattening features.
importance	variable importance mode. See ranger:rangerranger::ranger.
splitrule	Splitting rule. See ranger:rangerranger::ranger.

Value

An initialized DFNET.forest.

pievos101/DFNET documentation built on Dec. 1, 2022, 3:44 p.m.