diff options
Diffstat (limited to 'contrib/lua-torch/decisiontree/DecisionTree.lua')
| -rw-r--r-- | contrib/lua-torch/decisiontree/DecisionTree.lua | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/contrib/lua-torch/decisiontree/DecisionTree.lua b/contrib/lua-torch/decisiontree/DecisionTree.lua new file mode 100644 index 000000000..c61bc3757 --- /dev/null +++ b/contrib/lua-torch/decisiontree/DecisionTree.lua @@ -0,0 +1,12 @@ +local dt = require "decisiontree._env" + +-- An interface for decision trees. +local DecisionTree = torch.class("dt.DecisionTree", dt) + +-- Score an input example and return the prediction score. +-- input is a Tensor or SparseTensor +-- return prediction score and nodeId +function DecisionTree:score(input) + error"Not Implemented" + return score, nodeId +end |