python_Code/threshold.py

import spacy

def set_threshold(model_path, threshold):
    # Load the trained model
    nlp = spacy.load(model_path)

    # Set the threshold for text classification
    nlp.get_pipe('textcat_multilabel').threshold = threshold

    return nlp

# Example usage:
if __name__ == "__main__":
    model_path = "./my_trained_model"
    threshold = 0.21
    nlp = set_threshold(model_path, threshold)