← Về thư mục
skills/deepchem/scripts/transfer_learning.py
- train_chemberta · function · L37-L97 — def train_chemberta(train_dataset, valid_dataset, test_dataset, task_type='classification', n_tasks=1, n_epochs=10)
- train_grover · function · L100-L156 — def train_grover(train_dataset, test_dataset, task_type='classification', n_tasks=1, n_epochs=20)
- train_molformer · function · L159-L217 — def train_molformer(train_dataset, valid_dataset, test_dataset, task_type='classification', n_tasks=1, n_epochs=10)
- load_molnet_dataset · function · L220-L261 — def load_molnet_dataset(dataset_name, model_type)
- load_custom_dataset · function · L264-L310 — def load_custom_dataset(data_path, target_cols, smiles_col, model_type)
- main · function · L313-L439 — def main()