Re-organize GeMOSA

2021-05-27 15:44:01 +08:00
parent 8961215416
commit 6da60664f5
10 changed files with 354 additions and 350 deletions
--- a/xautodl/datasets/synthetic_env.py
+++ b/xautodl/datasets/synthetic_env.py
@@ -49,6 +49,10 @@ class SyntheticDEnv(data.Dataset):
        self._meta_info["task"] = "classification"
        self._meta_info["num_classes"] = int(num_classes)

+    @property
+    def oracle_map(self):
+        return self._oracle_map
+
    @property
    def meta_info(self):
        return self._meta_info