db14ddbf98a77ef1248b123ef8a37231973abdf3,finetune/datasets/reuters.py,,,#,72

Before Change

        test_size=0.2,
        random_state=42
    )
    model = SequenceLabeler(batch_size=1, n_epochs=3, val_size=0.0, max_length=512, chunk_long_sequences=True, subtoken_predictions=False, crf_sequence_labeling=True, multi_label_sequences=False)
    model.fit(trainX, trainY)
    predictions = model.predict(testX)
    print(predictions)
    print(annotation_report(testY, predictions))

After Change

        test_size=0.2,
        random_state=42
    )
    model = SequenceLabeler(base_model=base_model, n_epochs=3)
    model.fit(trainX, trainY)
    predictions = model.predict(testX)
    print(predictions)
    print(annotation_report(testY, predictions))

Instances