Train and analyze a classifier | OpenAI Interview Question