Skip to content

Latest commit

 

History

History
16 lines (12 loc) · 756 Bytes

README.md

File metadata and controls

16 lines (12 loc) · 756 Bytes

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.