参考:https://github.com/HannesStark/3DInfomax 1、3DInfomax 化合物分子基于对比学习的向量表示 首先把相关下载下来: git clone https://github.com/HannesStark/3DInfomax.git 1)数据分子准备和yml文件创建 ##数据加载 1万个smi分子,文件下载可参考:https://github.com/zilliztech/MolSearch/edit/master/script/test_1w.smi mols2 = [] with open