Add get_classifier function to classification