ikki407 / stacking

Stacked Generalization (Ensemble Learning)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

For more simple model wrapper class

ikki407 opened this issue · comments

All models are described in base.py.

For more simplicity, each model wrapper(XGB, Keras, VW, ...) should be described in each module, i.e., base_XGB.py, base_keras.py.

Then, we use in base.py like:

import base_XGB

class XGBClassifier(base_XGB.XGBClassifier):
    def __init__(self, params={}, num_round=50):
        base_XGB.XGBClassifier.__init__(self, params=params, num_round=num_round)