b6580cf7fae50ad9b954e2bde45cca7883fff2d3,examples/qm7/qm7_datasets.py,,load_qm7b_from_mat,#Any#Any#,57

Before Change

def load_qm7b_from_mat(featurizer=None, split="stratified"):

  if not os.path.exists("qm7b.mat"): os.system("wget http://www.quantum-machine.org/data/qm7b.mat")
  dataset = scipy.io.loadmat("qm7b.mat")
  
  X = dataset["X"]

After Change

  return qm7_tasks, (train_dataset, valid_dataset, test_dataset), transformers

def load_qm7b_from_mat(featurizer=None, split="stratified"):
  current_dir = os.path.dirname(os.path.realpath(__file__))
  dataset_file = os.path.join(
      current_dir, "./qm7b.mat")
  
  if not os.path.exists(dataset_file): os.system("wget -P "+current_dir+" http://www.quantum-machine.org/data/qm7b.mat")
  dataset = scipy.io.loadmat(dataset_file)
 
  X = dataset["X"]

Instances