Spaces:
Sleeping
Sleeping
saicharan2804
commited on
Commit
·
20631f9
1
Parent(s):
effa38c
Updated
Browse files- molgenevalmetric.py +3 -3
molgenevalmetric.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import evaluate
|
2 |
import datasets
|
3 |
# import moses
|
4 |
-
|
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 |
-
|
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)
|