blob: c61bc37577db64671d33e4ccc02f6285c2fa462e (
plain)
1
2
3
4
5
6
7
8
9
10
11
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
|