Solvve / ml_thematic_nlg

Thematic NLG

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPT-2 text generator

License Python 3.7 scikit-learn 0.23.2 Solvve

Description

Text generator for amazon polarity comments, using GPT-2 and aitextgen algorithms with tuning techniques.

We follow the next steps:

  1. EDA
  2. Data preprocessing
  3. Modeling

Dataset contains around 36000000 rows of polarity comments

Dataset

1.https://huggingface.co/datasets/amazon_polarity

About

Thematic NLG

License:MIT License


Languages

Language:Jupyter Notebook 100.0%