Commit 1f2102f0 authored by Vignesh's avatar Vignesh
Browse files

Added UV tasks

parent 0fad0a22
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -12,6 +12,7 @@ import time
import numpy as np
import deepchem
from deepchem.molnet.load_function.kaggle_features import merck_descriptors
from deepchem.molnet.load_function.uv_tasks import UV_tasks

logger = logging.getLogger(__name__)

@@ -144,8 +145,6 @@ def load_uv(shard_size=2000, featurizer=None, split=None, reload=True):
  data_dir = deepchem.utils.get_data_dir()
  data_dir = os.path.join(data_dir, "UV")

  UV_tasks = None

  if not os.path.exists(data_dir):
    os.mkdir(data_dir)

+28 −0
Original line number Diff line number Diff line
UV_tasks = ['w__210', 'w__211', 'w__212', 'w__213', 'w__214', 'w__215', 'w__216',
            'w__217', 'w__218', 'w__219', 'w__220', 'w__221', 'w__222', 'w__223',
            'w__224', 'w__225', 'w__226', 'w__227', 'w__228', 'w__229', 'w__230',
            'w__231', 'w__232', 'w__233', 'w__234', 'w__235', 'w__236', 'w__237',
            'w__238', 'w__239', 'w__240', 'w__241', 'w__242', 'w__243', 'w__244',
            'w__245', 'w__246', 'w__247', 'w__248', 'w__249', 'w__250', 'w__251',
            'w__252', 'w__253', 'w__254', 'w__255', 'w__256', 'w__257', 'w__258',
            'w__259', 'w__260', 'w__261', 'w__262', 'w__263', 'w__264', 'w__265',
            'w__266', 'w__267', 'w__268', 'w__269', 'w__270', 'w__271', 'w__272',
            'w__273', 'w__274', 'w__275', 'w__276', 'w__277', 'w__278', 'w__279',
            'w__280', 'w__281', 'w__282', 'w__283', 'w__284', 'w__285', 'w__286',
            'w__287', 'w__288', 'w__289', 'w__290', 'w__291', 'w__292', 'w__293',
            'w__294', 'w__295', 'w__296', 'w__297', 'w__298', 'w__299', 'w__300',
            'w__301', 'w__302', 'w__303', 'w__304', 'w__305', 'w__306', 'w__307',
            'w__308', 'w__309', 'w__310', 'w__311', 'w__312', 'w__313', 'w__314',
            'w__315', 'w__316', 'w__317', 'w__318', 'w__319', 'w__320', 'w__321',
            'w__322', 'w__323', 'w__324', 'w__325', 'w__326', 'w__327', 'w__328',
            'w__329', 'w__330', 'w__331', 'w__332', 'w__333', 'w__334', 'w__335',
            'w__336', 'w__337', 'w__338', 'w__339', 'w__340', 'w__341', 'w__342',
            'w__343', 'w__344', 'w__345', 'w__346', 'w__347', 'w__348', 'w__349',
            'w__350', 'w__351', 'w__352', 'w__353', 'w__354', 'w__355', 'w__356',
            'w__357', 'w__358', 'w__359', 'w__360', 'w__361', 'w__362', 'w__363',
            'w__364', 'w__365', 'w__366', 'w__367', 'w__368', 'w__369', 'w__370',
            'w__371', 'w__372', 'w__373', 'w__374', 'w__375', 'w__376', 'w__377',
            'w__378', 'w__379', 'w__380', 'w__381', 'w__382', 'w__383', 'w__384',
            'w__385', 'w__386', 'w__387', 'w__388', 'w__389', 'w__390', 'w__391',
            'w__392', 'w__393', 'w__394', 'w__395', 'w__396', 'w__397', 'w__398',
            'w__399', 'w__400']