You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

DecisionTree.lua 356B

123456789101112
  1. local dt = require "decisiontree._env"
  2. -- An interface for decision trees.
  3. local DecisionTree = torch.class("dt.DecisionTree", dt)
  4. -- Score an input example and return the prediction score.
  5. -- input is a Tensor or SparseTensor
  6. -- return prediction score and nodeId
  7. function DecisionTree:score(input)
  8. error"Not Implemented"
  9. return score, nodeId
  10. end