def train_svm(): data_helper = DataHelper() train_text, train_labels, ver_text, ver_labels, test_text, test_labels = data_helper.get_data_and_labels() stopwords = data_helper.get_stopwords() svm = SVM(train_text, train_labels, ver_text, ver_labels, test_text, test_labels, stopwords) svm.train() svm.verification() print('ver_acc: {:.3}'.format(svm.ver_acc)) svm.test() print('test_acc: {:.3}'.format(svm.test_acc))