5307ae3f7d1c4b6bd2285f946db82d944f3844db,examples/text_classification/tutorial_imdb_fasttext.py,,train_test_and_save_model,#,123

Before Change

        )
        print("Test accuracy: %.5f" % test_accuracy)

        classifier.save(sess, MODEL_FILE_PATH)


if __name__ == "__main__":
    train_test_and_save_model()

After Change

        for epoch in range(N_EPOCH):
            start_time = time.time()
            print("Epoch %d/%d" % (epoch + 1, N_EPOCH))
            train_accuracy = list()
            for X_batch, y_batch in tl.iterate.minibatches(X_train, y_train, batch_size=BATCH_SIZE, shuffle=True):
                accuracy, _ = sess.run(
                    [classifier.accuracy, classifier.train_op], feed_dict={
                        classifier.inputs: tl.prepro.pad_sequences(X_batch),
                        classifier.labels: y_batch,
                    }
                )
                train_accuracy.append(accuracy)
                if len(train_accuracy) % N_STEPS_TO_PRINT == 0:
                    print("\t[%d/%d][%d]accuracy " % (epoch + 1, N_EPOCH, len(train_accuracy)),
                          np.mean(train_accuracy[-N_STEPS_TO_PRINT:]))

Instances