From 4dd803775e4739ef8071627c1c650cd8e3455ae5 Mon Sep 17 00:00:00 2001 From: Katsuhiko Ishiguro Date: Mon, 6 Apr 2020 16:26:46 +0900 Subject: [PATCH] Validated by test runs on GPU/Kubernetes, LGTM --- .../dataset/converters/__init__.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/chainer_chemistry/dataset/converters/__init__.py b/chainer_chemistry/dataset/converters/__init__.py index eba6e92c..cf0d5ebc 100644 --- a/chainer_chemistry/dataset/converters/__init__.py +++ b/chainer_chemistry/dataset/converters/__init__.py @@ -6,16 +6,34 @@ 'ecfp': concat_mols, 'nfp': concat_mols, 'nfp_gwm': concat_mols, + 'nfp_wle': concat_mols, + 'nfp_cwle': concat_mols, + 'nfp_gwle': concat_mols, 'ggnn': concat_mols, 'ggnn_gwm': concat_mols, + 'ggnn_wle': concat_mols, + 'ggnn_cwle': concat_mols, + 'ggnn_gwle': concat_mols, 'gin': concat_mols, 'gin_gwm': concat_mols, + 'gin_wle': concat_mols, + 'gin_cwle': concat_mols, + 'gin_gwle': concat_mols, 'schnet': concat_mols, 'weavenet': concat_mols, 'relgcn': concat_mols, + 'relgcn_wle': concat_mols, + 'relgcn_cwle': concat_mols, + 'relgcn_gwle': concat_mols, 'rsgcn': concat_mols, 'rsgcn_gwm': concat_mols, + 'rsgcn_wle': concat_mols, + 'rsgcn_cwle': concat_mols, + 'rsgcn_gwle': concat_mols, 'relgat': concat_mols, + 'relgat_wle': concat_mols, + 'relgat_cwle': concat_mols, + 'relgat_gwle': concat_mols, 'gnnfilm': concat_mols, 'megnet': megnet_converter, 'cgcnn': cgcnn_converter