Create space textcategorizer model

email-spam-classifier-create-model.py

######## Main method ########

def main():

    # Load dataset
    data = pd.read_csv(data_path)
    observations = len(data.index)
    print("Dataset Size: {}".format(observations))

    # Create an empty spacy model
    nlp = spacy.blank("en")

    # Create the TextCategorizer with exclusive classes and "bow" architecture
    text_cat = nlp.create_pipe(
                  "textcat",
                  config={
                    "exclusive_classes": True,
                    "architecture": "bow"})

    # Adding the TextCategorizer to the created empty model
    nlp.add_pipe(text_cat)

    # Add labels to text classifier
    text_cat.add_label("ham")
    text_cat.add_label("spam")


if __name__ == "__main__":
    main()