Research Article
Hotel Review Classification Based on the Text Pretraining Heterogeneous Graph Neural Network Model
| Dataset | Comparative model | Accuracy | Recall | F1 score |
| Hotel review (our dataset) | BERT + HGAN | 0.70 | 0.72 | 0.72 | BERT | 0.62 | 0.62 | 0.61 | HGAN | 0.51 | 0.46 | 0.59 | ERNIE2.0 | 0.69 | 0.67 | 0.68 | TF-IDF + SVM | 0.36 | 0.30 | 0.33 | TextCNN | 0.50 | 0.50 | 0.48 |
| THUCNews | BERT + HGAN | 0.83 | 0.84 | 0.84 | BERT | 0.72 | 0.74 | 0.68 | HGAN | 0.71 | 0.75 | 0.73 | ERNIE2.0 | 0.78 | 0.74 | 0.74 | TF-IDF + SVM | 0.48 | 0.66 | 0.45 | TextCNN | 0.57 | 0.83 | 0.68 |
| Headlines Today News | BERT + HGAN | 0.78 | 0.76 | 0.80 | BERT | 0.65 | 0.83 | 0.62 | HGAN | 0.72 | 0.80 | 0.68 | ERNIE2.0 | 0.76 | 0.88 | 0.77 | TF-IDF + SVM | 0.58 | 0.67 | 0.62 | TextCNN | 0.65 | 0.72 | 0.73 |
|
|