modified CNN and added categorical-only
Jonathan Lee authored
f8d0ff16