from simpletransformers.classification import ClassificationModel, ClassificationArgs
model = ClassificationModel(‘xlm’, ‘seyonec/PubChem10M_SMILES_BPE_396_250’, args={‘evaluate_each_epoch’: True, ‘evaluate_during_training_verbose’: True, ‘no_save’: True, ‘num_train_epochs’: 10, ‘auto_weights’: True}) # You can set class weights by using the optional weight argument