happy transformer
Version 3.0.0
Документация и новости: Happytransformer.com
Присоединяйтесь к нашему Discord-серверу:
happy transformer позволяет легко настраивать и выполнять логические выводы с помощью моделей NLP Transformer.
О полном обновлении 3.0.0, включая критические изменения, читайте здесь.
Задачи | Вывод | Обучение |
---|---|---|
Генерация текста | ✔ | ✔ |
Классификация текста | ✔ | ✔ |
Предсказание слов | ✔ | ✔ |
Вопрос Ответ | ✔ | ✔ |
Преобразование текста в текст | ✔ | ✔ |
Прогнозирование следующего предложения | ✔ | |
Классификация токенов | ✔ |
pip install happytransformer
from happytransformer import HappyWordPrediction
#--------------------------------------#
happy_wp = HappyWordPrediction () # default uses distilbert-base-uncased
result = happy_wp . predict_mask ( "I think therefore I [MASK]" )
print ( result ) # [WordPredictionResult(token='am', score=0.10172799974679947)]
print ( result [ 0 ]. token ) # am
Генерация текста с обучением (GPT-Neo)
Классификация текста (обучение)
Классификация текста (обнаружение разжигания ненависти)
Классификация текста (анализ настроений)
Предсказание слов с обучением (DistilBERT, RoBERTa)
Лучшие модели Т5
Грамматическая коррекция
Точная настройка модели исправления грамматики