Partially fix dnn with binary classification