Research Article

Hotel Review Classification Based on the Text Pretraining Heterogeneous Graph Neural Network Model

Table 2

Comparison of results.

DatasetComparative modelAccuracyRecallF1 score

Hotel review (our dataset)BERT + HGAN0.700.720.72
BERT0.620.620.61
HGAN0.510.460.59
ERNIE2.00.690.670.68
TF-IDF + SVM0.360.300.33
TextCNN0.500.500.48

THUCNewsBERT + HGAN0.830.840.84
BERT0.720.740.68
HGAN0.710.750.73
ERNIE2.00.780.740.74
TF-IDF + SVM0.480.660.45
TextCNN0.570.830.68

Headlines Today NewsBERT + HGAN0.780.760.80
BERT0.650.830.62
HGAN0.720.800.68
ERNIE2.00.760.880.77
TF-IDF + SVM0.580.670.62
TextCNN0.650.720.73