saicharan2804 commited on
Commit
bbe40ff
1 Parent(s): 4baa95c

reverting back

Browse files
Files changed (1) hide show
  1. my_metric.py +13 -7
my_metric.py CHANGED
@@ -56,17 +56,23 @@ _CITATION = """
56
 
57
 
58
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
59
- class my_metric(evaluate.Measurement):
60
  def _info(self):
61
- return evaluate.MeasurementInfo(
62
- module_type="measurement",
63
  description=_DESCRIPTION,
64
  citation=_CITATION,
65
  inputs_description=_KWARGS_DESCRIPTION,
66
- features=datasets.Features({
67
- "generated_smiles": datasets.Value("string"),
68
- "train_smiles": datasets.Value("string"),
69
- }),
 
 
 
 
 
 
 
70
 
71
  reference_urls=["https://github.com/molecularsets/moses", "https://tdcommons.ai/functions/oracles/"],
72
  )
 
56
 
57
 
58
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
59
+ class my_metric(evaluate.Metric):
60
  def _info(self):
61
+ return evaluate.MetricInfo(
 
62
  description=_DESCRIPTION,
63
  citation=_CITATION,
64
  inputs_description=_KWARGS_DESCRIPTION,
65
+ features=datasets.Features(
66
+ {
67
+ "generated_smiles": datasets.Sequence(datasets.Value("string")),
68
+ "train_smiles": datasets.Sequence(datasets.Value("string")),
69
+ }
70
+ if self.config_name == "multilabel"
71
+ else {
72
+ "generated_smiles": datasets.Value("string"),
73
+ "train_smiles": datasets.Value("string"),
74
+ }
75
+ ),
76
 
77
  reference_urls=["https://github.com/molecularsets/moses", "https://tdcommons.ai/functions/oracles/"],
78
  )