ENH: write out classification network too