saicharan2804 commited on
Commit
effa38c
·
1 Parent(s): a764aef
Files changed (1) hide show
  1. molgenevalmetric.py +5 -3
molgenevalmetric.py CHANGED
@@ -1,7 +1,7 @@
1
  import evaluate
2
  import datasets
3
- import moses
4
- from moses import metrics
5
  import pandas as pd
6
  from tdc import Evaluator
7
  from tdc import Oracle
@@ -79,10 +79,12 @@ class molgenevalmetric(evaluate.Metric):
79
 
80
  def _compute(self, generated_smiles, train_smiles = None):
81
 
82
- Results = metrics.get_all_metrics(gen = generated_smiles, train= train_smiles)
83
 
84
  # evaluator = Evaluator(name = 'Diversity')
85
  # Diversity = evaluator(generated_smiles)
 
 
86
 
87
  evaluator = Evaluator(name = 'KL_Divergence')
88
  KL_Divergence = evaluator(generated_smiles, train_smiles)
 
1
  import evaluate
2
  import datasets
3
+ # import moses
4
+ # from moses import metrics
5
  import pandas as pd
6
  from tdc import Evaluator
7
  from tdc import Oracle
 
79
 
80
  def _compute(self, generated_smiles, train_smiles = None):
81
 
82
+ # Results = metrics.get_all_metrics(gen = generated_smiles, train= train_smiles)
83
 
84
  # evaluator = Evaluator(name = 'Diversity')
85
  # Diversity = evaluator(generated_smiles)
86
+
87
+ Results = {}
88
 
89
  evaluator = Evaluator(name = 'KL_Divergence')
90
  KL_Divergence = evaluator(generated_smiles, train_smiles)