e90d0bb446ad6da9d17ba1ab2d3fe6365e152961,finetune/datasets/stanford_sentiment_treebank.py,,,#,44

Before Change

if __name__ == "__main__":
    // Train and evaluate on SST
    dataset = StanfordSentimentTreebank(nrows=1000).dataframe
    model = Classifier(
        debugging_logs=True,
        interpolate_pos_embed=False, 
        n_epochs=3,
        batch_size=2,
        lr_warmup=0.1,
        max_length=64,
        base_model=GPTModel
    )
    trainX, testX, trainY, testY = train_test_split(dataset.Text.values, dataset.Target.values, test_size=0.3, random_state=42)
    model.fit(trainX, trainY)
    preds = model.predict(testX)
    print(preds, testY)

After Change

if __name__ == "__main__":
    // Train and evaluate on SST
    dataset = StanfordSentimentTreebank(nrows=1000).dataframe
    model = Classifier(
        debugging_logs=True,
        summarize_grads=True,
//        val_interval=1000,
    )
    trainX, testX, trainY, testY = train_test_split(dataset.Text.values, dataset.Target.values, test_size=0.3, random_state=42)
    model.fit(trainX, trainY)
    preds = model.predict(testX)
    print(preds, testY)

Instances