Skip to content

Solvve/ml_thematic_nlg

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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