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.