Skip to content

wearetyomsmnv/berterpretation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

berterpretation

Погрузитесь в глубины BERT с помощью мощных инструментов интерпретации.

ОсобенностиУстановкаИспользованиеТребованияСтруктураЛицензия


🌟 Особенности

  • 🧠 Анализ BERT: Загрузка и использование предобученных моделей BERT
  • 💬 Интерактивный режим: Анализ текстовых сообщений в реальном времени
  • 👁️ Визуализация внимания: Наглядное представление механизма внимания BERT
  • 🔍 Integrated Gradients: Глубокий анализ важности токенов
  • 🛠️ Интеграция с LIT: Расширенный анализ с помощью Language Interpretability Tool
  • ! Длина тензоров: 512

🚀 Установка

  1. Клонирование репозитория:

    git clone https://github.com/wearetyomsmnv/berterpretation/
    cd berterpretation
  2. Создание виртуального окружения:

    python -m venv venv
    source venv/bin/activate  # На Windows: venv\Scripts\activate
  3. Установка зависимостей:

    pip install -r requirements.txt
  4. Настройка меток и цветов для них:

Каждая модель умеет классифицровать по определённым меткам, их необходимо прописать в коде перед запуском скрипта:

src/scripts.py
valid_labels = ["about", "........ добавьте свои"]
  • в этой строке укажите свои метки

Также, вы можете определить цвета для меток:

src/scripts.py
color_map = {
 'humor': Fore.CYAN,
 'своя_метка': Fore.YELLOW,
 'своя_метка': Fore.GREEN,
 'своя_метка': Fore.LIGHTGREEN_EX,
 'своя_метка': Fore.LIGHTBLUE_EX,
 'своя_метка': Fore.LIGHTCYAN_EX,
 'своя_метка': Fore.LIGHTYELLOW_EX,
 'своя_метка': Fore.LIGHTMAGENTA_EX,
 'своя_метка': Fore.LIGHTRED_EX,
 'своя_метка': Fore.LIGHTWHITE_EX,
 'своя_метка': Fore.MAGENTA,
 'своя_метка': Fore.LIGHTMAGENTA_EX,
 'своя_метка': Fore.RED,
 'своя_метка': Fore.LIGHTRED_EX,
 'своя_метка': Fore.BLUE,
 'своя_метка': Fore.LIGHTBLUE_EX,
 'своя_метка': Fore.CYAN,
 'своя_метка': Fore.LIGHTCYAN_EX,
 'своя_метка': Fore.BLUE,
 'своя_метка': Fore.WHITE,
 'своя_метка': Fore.RED,
 'своя_метка': Fore.LIGHTRED_EX
 }
  • меняем на свои метки, тем самым окрашивая их

🖥️ Использование

image

Интерактивный режим

python main.py --interactive --model path/to/your/model

image

Проверки безопасности

python main.py --interactive --model path/to/your/model --check_security

image

Режим LIT

python main.py --lit --model path/to/your/model

Выбор устройства

python main.py --device [gpu,cpu]

📋 Требования

  • Python 3.7+
  • PyTorch
  • Transformers
  • Captum
  • Matplotlib
  • textattack
Полный список зависимостей

См. файл requirements.txt

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

berterpretation/
├── src/               # Скрипты
├── main.py            # Основной скрипт
├── README.md          # Документация
└── requirements.txt   # Зависимости

📄 Лицензия

Этот проект лицензирован под MIT License.

🤝 Вклад в проект

Мы приветствуем вклад сообщества! Если у вас есть идеи для улучшения:

  1. Форкните репозиторий
  2. Создайте свою ветку (git checkout -b feature/AmazingFeature)
  3. Зафиксируйте изменения (git commit -m 'Add some AmazingFeature')
  4. Отправьте изменения (git push origin feature/AmazingFeature)
  5. Откройте Pull Request

👨‍💻 Авторы


Создано с ❤️, wearetyomsmnv.

Releases

No releases published

Packages

No packages published

Languages