Другоеilyali034
ruT5-base-smart-home
русские команды в JSON для Home Assistant
1 мин чтенияобновлено
Это файнтюн ai-forever/ruT5-base (~220M параметров) от автора ilyali034. Модель принимает команду на русском для умного дома и возвращает JSON-список вызовов функций в стиле Home Assistant: HassTurnOn, HassTurnOff, HassLightSet, HassClimateSetTemperature и другие.
⚙ Ключевые особенности
- 💬 Понимает русские команды и выдаёт JSON со списком function calls.
- 🔒 Токенизатору не хватает символов
{ } Q Y Z & é →, поэтому нужен escaped-формат;inference.pyуже всё обрабатывает. - 🆗 Валидный JSON на тесте — 100%.
- 🎯 Exact match по каноническому JSON — 0.828; accuracy главной функции — 0.997.
- ❗ Почти все ошибки — в аргументе
name(имя устройства). - 🔎 Обучена на синтетическом датасете
RockMan256/needle-ru-opt.
💻 Пример
from huggingface_hub import snapshot_download
import sys
path = snapshot_download("ilyali034/ruT5-base-smart-home")
sys.path.insert(0, path)
from inference import SmartHomeModel
model = SmartHomeModel(path)
print(model.predict("включи свет на кухне"))Подробнее
Тест: 1290 примеров; exact match 0.828, valid JSON 1.000, function name F1 0.993, arguments accuracy 0.792.Обучение: 5 эпох, Adafactor, lr 3e-4, batch size 16, fp16; данные синтетические, split 90/5/5.
Ограничения: модель не видит список устройств, может ошибаться в имени устройства и иногда пропускает последний вызов; команды русские, имена устройств в данных в основном английские.
Комментарии