classify.py 文件源码

def create_model(clf_name, features, groundtruths, outdir, classifiers):
    begin = int(round(time.time() * 1000))
    utils.print_success("Starting " + clf_name)
    clf_dir = outdir + clf_name + "/"
    utils.create_dir(clf_dir)
    clf = classifiers[clf_name]
    clf.fit(features, groundtruths)
    joblib.dump(clf, clf_dir + clf_name + ".pkl")
    utils.print_info(clf_name + " done in " + str(int(round(time.time() * 1000)) - begin) + "ms")