-
Notifications
You must be signed in to change notification settings - Fork 1
Trainer
Luiz Miguel edited this page Sep 27, 2023
·
7 revisions
A simple trainer to store the patterns in your model.
model (Model, optional): The model that will be used for training, which may or may not exist. Defaults to Model().
train(action, entities: defaultdict | dict)
Trains the model using the inserted patterns
readPatterns(path: str, marker: str = "*")
Reads a .txt document with training patterns
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.