Skip to content

About

This script uses the THUCNews datasets to finetune the ERNIE model, with paddle and paddlehub.

Topics

Resources

Stars

1 star

Watchers

1 watching

Forks

Latest commit

 

History

3 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 

Repository files navigation

News-Classification-ERNIE

This script uses the THUCNews datasets to finetune ERNIE model, with paddle and paddlehub.

it is a project for a class assignment.

Several tasks are needed to be completed in the codes:

  1. display the version information of both paddle and paddlehub;
  2. print the first 3 and last 3 samples in the validation dataset of THUCNews;
  3. finetune the ERNIE(Enhanced Representation from kNowledge IntEgration) model; a. use the THUCNews datasets (https://aistudio.baidu.com/aistudio/datasetdetail/16287); b. display the model information of ERNIE model; c. conduct transfer learning with the model and the dataset; d. predict with the test dataset and several example.

Please feel free to contact me if you have any question.

About

This script uses the THUCNews datasets to finetune the ERNIE model, with paddle and paddlehub.

Topics

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages