| 85 | """ |
| 86 | |
| 87 | def _info(self): |
| 88 | return tfds.core.DatasetInfo( |
| 89 | builder=self, |
| 90 | description=_DESCRIPTION, |
| 91 | features=tfds.features.FeaturesDict({ |
| 92 | "name": tfds.features.Text(), # patient info |
| 93 | "image": tfds.features.Image(), |
| 94 | "label": tfds.features.Sequence( |
| 95 | tfds.features.ClassLabel(names=_LABELS.values()) |
| 96 | ), |
| 97 | "image_view": tfds.features.ClassLabel( |
| 98 | names=["frontal", "lateral"] |
| 99 | ), |
| 100 | }), |
| 101 | supervised_keys=("image", "label"), |
| 102 | homepage="https://stanfordmlgroup.github.io/competitions/chexpert/", |
| 103 | citation=_CITATION, |
| 104 | ) |
| 105 | |
| 106 | def _split_generators(self, dl_manager): |
| 107 | """Returns SplitGenerators.""" |