Skip to content

kexi-bq/embedding-explainer

Repository files navigation

🧠 Embedding GUI Editor

Интерактивный инструмент для визуализации, анализа и редактирования координат эмбеддингов текста.


🚀 Что делает

  • Получает эмбеддинг текста с помощью SentenceTransformer
  • Выявляет важные координаты, связанные с категориями: emotion, fact, question
  • Позволяет вручную изменять координаты: обнуление, инверсия, ослабление
  • Строит график сравнения исходного и модифицированного эмбеддинга
  • Предсказывает категорию и подкатегорию смысла (через RandomForest)
  • Имеет автоматический режим подбора координат для смены подкатегории

❓ Зачем это нужно

Эмбеддинги — это векторное представление "мыслей" модели.
Этот инструмент позволяет:

  • Понять, как модель "думает" о тексте
  • Объяснить поведение модели на векторном уровне
  • Редактировать смысл фразы без изменения самого текста

📦 Структура проекта

  • embedding_gui_editor.py — интерфейс на Streamlit
  • classifier.py — классификатор категорий и подкатегорий
  • dataset.csv — датасет фраз и меток
  • requirements.txt — зависимости проекта

⚙️ Установка и запуск

pip install -r requirements.txt
streamlit run embedding_gui_editor.py

About

Interactive editor for text meaning via embedding vector control

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published