def main(): # Fetch input parameters. args = return_args() test_input = args.test_data_url X, y, x = Data.load_data(args.training_data_file, test_input) # Plot data Plot.plot_data(X, y) # SVM model. model = SVMClassifier().train(X, y) # print training details print('\nmodel.X: {}\nmodel.y: {}\nmodel.alphas: {}\nmodel.w: {}\n'.format( model.X, model.y, model.alphas, model.w)) # prediction. prediction = SVMClassifier().classify(x, model) # print test details print('\nx: {}\nprediction: {}\n'.format(x, prediction)) # Plot boundary Plot.plot_boundary(X, y, x, model) print('\nProcessed... {}\n\nURL Classified as: {}\n'.format( test_input, 'wrong url' if prediction == 0 else 'correct url'))