fixing the classification: do not compare an integer with an array
fixing the classification: do not compare an integer with an array

Warvito/pydeeplearn,snurkabill/pydeeplearn,Warvito/pydeeplearn,mihaelacr/pydeeplearn,snurkabill/pydeeplearn