1mlpack_decision_tree(1) General Commands Manual mlpack_decision_tree(1)
2
6 mlpack_decision_tree - decision tree
7
9 mlpack_decision_tree [-h] [-v]
10
12 Train and evaluate using a decision tree. Given a dataset containing
13 numeric features and associated labels for each point in the dataset,
14 this program can train a decision tree on that data.
15 The training file and associated labels are specified with the --train‐
17 ing_file and --labels_file options, respectively. The labels should be
18 in the range [0, num_classes - 1].
19 When a model is trained, it may be saved to file with the --out‐
21 put_model_file (-M) option. A model may be loaded from file for predic‐
22 tions with the --input_model_file (-m) option. The --input_model_file
23 option may not be specified when the --training_file option is speci‐
24 fied. The --minimum_leaf_size (-n) parameter specifies the minimum num‐
25 ber of training points that must fall into each leaf for it to be
26 split. If --print_training_error (-e) is specified, the training error
27 will be printed.
28 A file containing test data may be specified with the --test_file (-T)
30 option, and if performance numbers are desired for that test set,
31 labels may be specified with the --test_labels_file (-L) option. Pre‐
32 dictions ffor each test point may be stored into the file specified by
33 the --predictions_file (-p) option. Class probabilities for each pre‐
34 diction will be stored in the file specified by the --probabili‐
35 ties_file (-P) option.
36
38 --help (-h)
39 Default help info.
40 --info [string]
42 Get help on a specific module or option. Default value ''.
43 --input_model_file (-m) [string] File to load pre-trained deci‐
44 sion tree from, to be used with test points. Default value ''.
45 --labels_file (-l) [string]
47 File containing training labels. Default value ’'. --mini‐
48 mum_leaf_size (-n) [int] Minimum number of points in a leaf.
49 Default value 20. --output_model_file (-M) [string] File to
50 save trained decision tree to. Default value ''. --predic‐
51 tions_file (-p) [string] File to save class predictions to for
52 each test point. Default value ''.
53 --print_training_error (-e)
55 Print the training error. --probabilities_file (-P) [string]
56 File to save class probabilities to for each test point. Default
57 value ''.
58 --test_file (-T) [string]
60 File containing test points. Default value ''.
61 --test_labels_file (-L) [string] File containing test labels, if
62 accuracy calculation is desired. Default value ''. --train‐
63 ing_file (-t) [string] File containing training points. Default
64 value ’'.
65 --verbose (-v)
67 Display informational messages and the full list of parameters
68 and timers at the end of execution.
69 --version (-V)
71 Display the version of mlpack.
72
75 For further information, including relevant papers, citations, and the‐
76 ory, For further information, including relevant papers, citations, and
77 theory, consult the documentation found at http://www.mlpack.org or
78 included with your consult the documentation found at
79 http://www.mlpack.org or included with your DISTRIBUTION OF MLPACK.
80 DISTRIBUTION OF MLPACK.
81 mlpack_decision_tree(1)