HaveAI commited on
Commit
aa13ff5
·
verified ·
1 Parent(s): cadf9c8

Upload 6 files

Browse files
data/train_data.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ [
2
+ {"input": "Привет", "output": "Привет! Как я могу помочь?"},
3
+ {"input": "Сколько будет 2+2?", "output": "2+2 будет 4."},
4
+ {"input": "Расскажи шутку", "output": "Почему программисты не любят природу? Потому что там много багов!"},
5
+ {"input": "Что такое искусственный интеллект?", "output": "Это область науки, которая занимается созданием машин и программ, способных выполнять задачи, требующие умственного труда человека."}
6
+ ]
flare_model.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import random
3
+ import json
4
+
5
+ class FlareModel:
6
+ def __init__(self):
7
+ self.context = {}
8
+
9
+ def analyze_message(self, message):
10
+ # Пример анализа сообщения
11
+ return f"Сообщение анализируется: {message}"
12
+
13
+ def generate_response(self, message):
14
+ responses = [
15
+ "Привет, как я могу помочь?",
16
+ "Что-то еще? Спрашивай!",
17
+ "А что ты думаешь о том, что я только что сказал?"
18
+ ]
19
+ return random.choice(responses)
20
+
21
+ def load_context(self, data):
22
+ self.context.update(data)
23
+
24
+ def get_context(self):
25
+ return self.context
flare_train.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import json
3
+ from flare_model import FlareModel
4
+
5
+ def train_flare(model, train_data_file):
6
+ with open(train_data_file, 'r') as file:
7
+ train_data = json.load(file)
8
+
9
+ for category, examples in train_data.items():
10
+ print(f"Обучение по категории: {category}")
11
+ for example in examples:
12
+ print(f"Пример: {example}")
13
+
14
+ if __name__ == "__main__":
15
+ flare_model = FlareModel()
16
+ train_flare(flare_model, 'train_data.json')
model/train_model.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+
3
+ def load_data():
4
+ with open('data/train_data.json', 'r', encoding='utf-8') as f:
5
+ return json.load(f)
6
+
7
+ def train_model(data):
8
+ for item in data:
9
+ print(f"Обучаем модель на вопрос: {item['input']} -> Ответ: {item['output']}")
10
+
11
+ def main():
12
+ data = load_data()
13
+ train_model(data)
14
+
15
+ if __name__ == "__main__":
16
+ main()
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ torch
2
+ transformers
train_data.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+
2
+ {
3
+ "greetings": ["Привет", "Здравствуй", "Добрый день"],
4
+ "tasks": ["Как дела?", "Что ты можешь делать?", "Какие у тебя способности?"],
5
+ "logic_tasks": ["2 + 2", "Что такое 5 умножить на 4?"]
6
+ }