Change classifier.py according to the current architecture.
Change classifier.py according to the current architecture.