saicharan2804 commited on
Commit
20631f9
·
1 Parent(s): effa38c
Files changed (1) hide show
  1. molgenevalmetric.py +3 -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,12 +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
- Results = {}
88
 
89
  evaluator = Evaluator(name = 'KL_Divergence')
90
  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)