Создайте классификатор настроений фильма с моделью Google Bert Language Model
Это пример создания классификатора рецензий фильмов с помощью Google Bert (двунаправленные представления энкодера от Transformers). Модель языка NLP.
Этот код требует scikit-learn , tensorflow-gpu , tensorflow-hub , bert-tensorflow . Код совместен с TF <= 1.1.50 и последней доступной моделью BERT в Tensorflow Hub. Чтобы использовать версию процессора, установите tensorflow==1.15.0 .
pip install scikit-learn
pip install tensorflow-gpu==1.15.0
pip install tensorflow-hub
pip install bert-tensorflowЧтобы запустить этот проект, вы можете
Откройте ноутбук ipython src/bert_sentiment_classifier-local.ipynb в ноутбуке Juypter или
Импорт src/bert_sentiment_classifier.ipynb в Colab Google с бэкэнд GPU.
Откройте Pyhon Interactive src/bert_sentiment_classifier.py в коде VisualStudio. Смотрите здесь, как это работает с ноутбуками и кодом Jupyter.