Spaces:
Sleeping
Sleeping
saicharan2804
commited on
Commit
·
effa38c
1
Parent(s):
a764aef
Check
Browse files- 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)
|