Пример #1
0
def main():
    Xtrain, Ytrain, Xtest, Ytest, word2idx, tag2idx = get_data()

    V = len(word2idx)
    print("vocabulary size:", V)
    K = len(tag2idx)

    # train and score
    model = LogisticRegression()
    model.fit(Xtrain, Ytrain, V=V, K=K, epochs=5)
    print("training complete")
    print("train score:", model.score(Xtrain, Ytrain))
    print("train f1 score:", model.f1_score(Xtrain, Ytrain))
    print("test score:", model.score(Xtest, Ytest))
    print("test f1 score:", model.f1_score(Xtest, Ytest))
def main():
    Xtrain, Ytrain, Xtest, Ytest, word2idx, tag2idx = get_data()

    V = len(word2idx)
    print "vocabulary size:", V
    K = len(tag2idx)

    # train and score
    model = LogisticRegression()
    model.fit(Xtrain, Ytrain, V=V, K=K, epochs=5)
    print "training complete"
    print "train score:", model.score(Xtrain, Ytrain)
    print "train f1 score:", model.f1_score(Xtrain, Ytrain)
    print "test score:", model.score(Xtest, Ytest)
    print "test f1 score:", model.f1_score(Xtest, Ytest)
#     return Xtrain, Ytrain, Xtest, Ytest, word2idx, tag2idx


def main():
    Xtrain, Ytrain, Xtest, Ytest, word2idx, tag2idx = get_data()

    V = len(word2idx)
<<<<<<< HEAD
    print "vocabulary size:", V
=======
    print("vocabulary size:", V)
>>>>>>> upstream/master
    K = len(tag2idx)

    # train and score
    model = LogisticRegression()
    model.fit(Xtrain, Ytrain, V=V, K=K, epochs=5)
<<<<<<< HEAD
    print "training complete"
    print "train score:", model.score(Xtrain, Ytrain)
    print "train f1 score:", model.f1_score(Xtrain, Ytrain)
    print "test score:", model.score(Xtest, Ytest)
    print "test f1 score:", model.f1_score(Xtest, Ytest)
=======
    print("training complete")
    print("train score:", model.score(Xtrain, Ytrain))
    print("train f1 score:", model.f1_score(Xtrain, Ytrain))
    print("test score:", model.score(Xtest, Ytest))
    print("test f1 score:", model.f1_score(Xtest, Ytest))
>>>>>>> upstream/master