train_classifier.py 文件源码

def learn(training_data, training_labels, show_score=False, store=False):

    print ("Start Learning....")

    clf = SVC(kernel='linear', probability=True, C=1)

    clf.fit(training_data, training_labels)

    print ("Done Learning.")


    if store:
        print ("Pickling classifier...")
        pickle.dump(clf, open(path_config.CLASSIFIER_PICKLING_FILE, 'wb'))
        print ("Done Pickling.")

    if show_score:
        print ("Scoring classifier ...")
        print ("Data-Level Training Set Prediction Accuracy: %s" % clf.score(training_data, training_labels))