Function main

hmm_class/hmm_classifier.py:88–99

()

        

def main():
    X, Y, V = get_data()
    print("len(X):", len(X))
    print("Vocabulary size:", V)
    X, Y = shuffle(X, Y)
    N = 20 # number to test
    Xtrain, Ytrain = X[:-N], Y[:-N]
    Xtest, Ytest = X[-N:], Y[-N:]

    model = HMMClassifier()
    model.fit(Xtrain, Ytrain, V)
    print("Score:", model.score(Xtest, Ytest))


if __name__ == '__main__':