b6580cf7fae50ad9b954e2bde45cca7883fff2d3,examples/qm7/qm7_datasets.py,,load_qm7_from_mat,#Any#Any#,15

Before Change

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

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

After Change

import csv

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

Instances