FlareNew / flare_train.py
HaveAI's picture
Upload 6 files
aa13ff5 verified
raw
history blame
484 Bytes
import json
from flare_model import FlareModel
def train_flare(model, train_data_file):
with open(train_data_file, 'r') as file:
train_data = json.load(file)
for category, examples in train_data.items():
print(f"Обучение по категории: {category}")
for example in examples:
print(f"Пример: {example}")
if __name__ == "__main__":
flare_model = FlareModel()
train_flare(flare_model, 'train_data.json')