ML之RF:利用Js语言设计随机森林算法【DT之CART算法(gain index)】&并应用随机森林算法
ML之RF:利用Js语言设计随机森林算法【DT之CART算法(gain index)】&并应用随机森林算法
输出结果
设计思路
代码实现(部分代码)
var doTest = function() {
option = {};
option.treeNumber = 100;
option.bagNumber = g.samples.length*0.80;
option.depth = 8;
option.bestSelect = 10;
var forest = createForest(g.samples, option);
var result = {};
for(var x = -23; x <= 23; x += 1) {
for(var y = -23; y <= 23; y += 1) {
var sample = [x/10, y/10];
var pred = predWithForest(forest, sample);
result[x/10 + ',' + y/10] = pred;
}
}
return result;
}
赞 (0)