『科技之感』提升ImageNet分类准确率且可解释,决策树的复兴?结合神经网络( 四 )


model=wrn28_10_cifar10()
model=SoftNBDT(
pretrained=True,
dataset='CIFAR10',
arch='wrn28_10_cifar10',
model=model)
另外 , 研究者还提供了如何用少于6行代码将nbdt与我们自己的神经网络相结合 , 详细内容请见其GitHub开源项目 。