随便看看AutoGluon-Tabular
发布时间
阅读量：
阅读量
论文中指出，需提前构建若干模型，并依据人工设定的优先级顺序开展训练工作。
# Higher values indicate higher priority, priority dictates the order models are trained for a given level.
DEFAULT_MODEL_PRIORITY = dict(
RF=100,
XT=90,
KNN=80,
GBM=70,
CAT=60,
NN=50,
LR=40,
custom=0,
)
感觉设定显得较为随意
MODEL_TYPES = dict(
RF=RFModel,
XT=XTModel,
KNN=KNNModel,
GBM=LGBModel,
CAT=CatboostModel,
NN=TabularNeuralNetModel,
LR=LinearModel,
)
全部评论 (0)
还没有任何评论哟~