Linear Classifier Python