rspamd/contrib/lua-torch/decisiontree/DecisionForestTrainer.lua

local dt = require "decisiontree._env"

-- Interface for all decisionForestTrainers
local DFT = torch.class("dt.DecisionForestTrainer", dt)

-- Train a DecisionForest with examples, a table of valid featureIds and a dataset (i.e. sortedExamplesByFeatureId)
function DFT:train(examples, validFeatureIds, dataset)
   assert(torch.type(examples) == "table")
   assert(torch.isTypeOf(examples[1], "dt.LabeledExample"))

   assert(torch.type(validFeatureIds) == 'table')

   assert(torch.type(dataset) == 'table')
   for k,v in pairs(dataset) do
      assert(torch.type(v) == 'table')
      assert(torch.isTypeOf(v[1], 'dt.LabeledExample'))
      break
   end
   -- dataset is a table mapping featureIds to sorted lists of LabeledExamples
   -- e.g. {featureId={example1,example2,example3}}
   error"Not Implemented"
end