For more simple model wrapper class
ikki407 opened this issue · comments
Ikki Tanaka commented
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)