用于构建复杂scikit学习管道的基于图的函数api。
baikal的Python项目详细描述
baikal是一个基于图的函数式api,用于构建复杂的机器学习 实现scikit学习api的对象管道。主要是灵感 关于用于深度学习的优秀Kerasapi 来自TensorFlow框架的一些概念 以及(也许鲜为人知)的graphkit 包裹。
baikal旨在提供一个允许构建复杂的、非线性的api 机器学习管道如下:
代码如下:
x1=Input()x2=Input()y1=ExtraTreesClassifier()(x1)y2=RandomForestClassifier()(x2)z=PowerTransformer()(x2)z=PCA()(z)y3=LogisticRegression()(z)ensemble_features=Stack()([y1,y2,y3])y=SVC()(ensemble_features)model=Model([x1,x2],y)
baikal与python>;=3.5兼容,分布在 BSD 3条款许可证。