saicharan2804 commited on
Commit
18551e5
·
1 Parent(s): 9ebf1eb

Adding synthetic_complexity_score

Browse files
Files changed (1) hide show
  1. molgenevalmetric.py +2 -1
molgenevalmetric.py CHANGED
@@ -74,7 +74,7 @@ class SCScorer():
74
  self.score_scale = score_scale
75
  self._restored = False
76
 
77
- def restore(self, weight_path='model.ckpt-10654.as_numpy.json.gz', FP_rad, FP_len):
78
  self.FP_len = FP_len
79
  self.FP_rad = FP_rad
80
  self._load_vars(weight_path)
@@ -100,6 +100,7 @@ class SCScorer():
100
  self._restored = True
101
  return self
102
 
 
103
  def smi_to_fp(self, smi):
104
  if not smi:
105
  return np.zeros((self.FP_len,), dtype=np.float32)
 
74
  self.score_scale = score_scale
75
  self._restored = False
76
 
77
+ def restore(self, weight_path=os.path.join('model.ckpt-10654.as_numpy.json.gz'), FP_rad=FP_rad, FP_len=FP_len):
78
  self.FP_len = FP_len
79
  self.FP_rad = FP_rad
80
  self._load_vars(weight_path)
 
100
  self._restored = True
101
  return self
102
 
103
+
104
  def smi_to_fp(self, smi):
105
  if not smi:
106
  return np.zeros((self.FP_len,), dtype=np.float32)