Skip to content

Trainer

Luiz Miguel edited this page Sep 27, 2023 · 7 revisions

A simple trainer to store the patterns in your model.

Args:

model (Model, optional): The model that will be used for training, which may or may not exist. Defaults to Model().

Methods

train(action, entities: defaultdict | dict) Trains the model using the inserted patterns
readPatterns(path: str, marker: str = "*") Reads a .txt document with training patterns

Exemple:

from simpleclassifier import Trainer
from collections import defaultdict

trainner = Trainer()
patterns = defaultdict(list)

while True:
    try:
        entity = input("Entity > ")
        inputText = input("> ")
        patterns[entity].append(inputText)
    except Exception:
        break
    finally:
        trainer.train(patterns)

Or

from simpleclassifier import Trainer

trainner = Trainer()

patterns = trainer.readPatterns("patterns.txt")
trainer.train(patterns)

To learn more about read patterns, click here.

Clone this wiki locally