Basic Linear Classifier Python Tutorial Gfg Problem