244c6805387c01208c0307fcf8895271074dde94,finetune/datasets/reuters.py,,,#,73

Before Change

    exit()
//    print(annotation_report(testY, predictions))
    model.config.float_16_predict=False
    model.config.predict_batch_size = 32
    predictions_fp16 = model.predict(testX * 100)
//    print(annotation_report(testY, predictions_fp16))
    exit()
    for x, x16 in zip(predictions, predictions_fp16):
        print(x)
        print(x16)
        input()

After Change

        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))
    sequence_labeling_token_confusion(testX, testY, predictions)

Instances