iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix GeMOSA's bugs

Browse Source
This commit is contained in:
D-X-Y
2021-05-27 01:09:28 -07:00
parent 4ca38ad2ff
commit 1ce0b80776
4 changed files with 32 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
xautodl/datasets/synthetic_env.py
View File

@@ -46,15 +46,19 @@ class SyntheticDEnv(data.Dataset):
def set_regression(self):
self._meta_info["task"] = "regression"
self._meta_info["input_dim"] = self._data_generator.ndim
self._meta_info["output_shape"] = self._oracle_map.output_shape(self._data_generator.output_shape())
self._meta_info['output_dim'] = int(np.prod(self._meta_info["output_shape"]))
self._meta_info["output_shape"] = self._oracle_map.output_shape(
self._data_generator.output_shape()
)
self._meta_info["output_dim"] = int(np.prod(self._meta_info["output_shape"]))
def set_classification(self, num_classes):
self._meta_info["task"] = "classification"
self._meta_info["input_dim"] = self._data_generator.ndim
self._meta_info["num_classes"] = int(num_classes)
self._meta_info["output_shape"] = self._oracle_map.output_shape(self._data_generator.output_shape())
self._meta_info['output_dim'] = int(np.prod(self._meta_info["output_shape"]))
self._meta_info["output_shape"] = self._oracle_map.output_shape(
self._data_generator.output_shape()
)
self._meta_info["output_dim"] = int(np.prod(self._meta_info["output_shape"]))
@property
def oracle_map(self):