「机器之心」决策树的复兴?结合神经网络,提升ImageNet分类准确率且可解释( 四 )


from nbdt.model import SoftNBDT
from nbdt.models import ResNet18, wrn28_10_cifar10, wrn28_10_cifar100, wrn28_10# use wrn28_10 for TinyImagenet200
model = wrn28_10_cifar10()
model = SoftNBDT(
pretrained=True,
dataset='CIFAR10',
arch='wrn28_10_cifar10',
model=model)
【「机器之心」决策树的复兴?结合神经网络,提升ImageNet分类准确率且可解释】另外 , 研究者还提供了如何用少于 6 行代码将 nbdt 与我们自己的神经网络相结合 , 详细内容请见其 GitHub 开源项目 。