from milk.supervised import svm from milk.supervised import tree from milk.supervised.classifier import normalise data = [[1, 2], [2, 3], [3, 4], [4, 6], [5, 8]] labels = ['A', 'A', 'B', 'B', 'B'] data_normalized = normalise(data) svm_model = svm.svm_train(data_normalized, labels) tree_model = tree.tree_learner(data_normalized, labels) test_data = [[1, 1], [2, 2], [3, 3]] svm_predictions = svm.svm_predict(svm_model, test_data) tree_predictions = tree.tree_predict(tree_model, test_data)