2f1f4adb504fd6a91027a091a4e5e911a4d7c905,finetune/datasets/mcdonalds_yelp_sentiment.py,,,#,39

Before Change

    model = MultiLabelClassifier(verbose=True, n_epochs=2)
    trainX, testX, trainY, testY = train_test_split(dataset.Text, dataset.Target, test_size=0.3, random_state=42)
    model.fit(trainX, trainY)
    accuracy = np.mean(model.predict(testX) == testY)
    print("Test Accuracy: {:0.2f}".format(accuracy))

After Change

    trainX, testX, trainY, testY = train_test_split(dataset.Text, dataset.Target, test_size=0.3, random_state=42)
    model.fit(trainX, trainY)
    for threshold in np.linspace(0, 1, 5):
        print("Threshold = {}".format(threshold))
        print(classification_report(
            model.label_encoder.transform(testY),
            model.label_encoder.transform(model.predict(testX, threshold=threshold))

Instances