Research Article

Mol-BERT: An Effective Molecular Representation with BERT for Molecular Property Prediction

Table 2

The fine-tuning hyperparameters.

ParameterValue/range

Learning rate1-5∼1-3
Batch size8
Epoch100
OptimizerAdam
Embedding dimension300
Size of dictionary13,325
Number of attention head6
Layers of fully connected neural network6