kaz-Anova / StackNet

StackNet is a computational, scalable and analytical Meta modelling framework
MIT License
1.32k stars 344 forks source link

StackNet usage quesiton #23

Closed sreepunith closed 7 years ago

sreepunith commented 7 years ago

If I am using 2 layer StackNet for a regression problem. Can I use a regression algorithm in the second layer ? The documentation says _"A train method needs at least a train_file and a paramsfile. It also needs at least two algorithms, and the and last layer must not contain a regressor unless the metric is auc and the problem is binary" Also are the prediction order preserved (rows) ? Or will the outputs be shuffled?

Thanks

kaz-Anova commented 7 years ago

I need to change the decocumentation for that - this was before regression was added as a task .

Yes you can have a regressor in the last layer if you specifytask=regression

The initial order of rows is preserved in the predictions.

sreepunith commented 7 years ago

Thank you.