mirror da
https://github.com/rspamd/rspamd.git
synced 2024-07-29 20:17:47 +02:00
23 righe
816 B
Lua
23 righe
816 B
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
|