コード例 #1
0
def main():
    print("Reading Input")
    (reviews, targets) = InputReader.run()

    # Uncomment this for faster testing.
    #(reviews, targets) = (reviews[:100], targets[:100])

    print("Building Vocabulary")
    vocabulary = Vocabulary(reviews, targets)
    print(f"Vocab size: {vocabulary.num_words}")
    print("Converting inputs")
    inputs = vocabulary.featurize(reviews)

    print("Beginning Training")
    runner = Runner(inputs, targets)
    for epoch in range(100):
        runner.run_epoch()