File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -33,7 +33,7 @@ def get_dataset(self):
3333 """Download dataset associated with task"""
3434 return datasets .get_dataset (self .dataset_id )
3535
36- def get_X_and_Y (self ):
36+ def get_X_and_y (self ):
3737 dataset = self .get_dataset ()
3838 # Replace with retrieve from cache
3939 if 'Supervised Classification' .lower () in self .task_type .lower ():
@@ -42,9 +42,9 @@ def get_X_and_Y(self):
4242 target_dtype = float
4343 else :
4444 raise NotImplementedError (self .task_type )
45- X_and_Y = dataset .get_data (target = self .target_feature ,
45+ X_and_y = dataset .get_data (target = self .target_feature ,
4646 target_dtype = target_dtype )
47- return X_and_Y
47+ return X_and_y
4848
4949 def get_train_test_split_indices (self , fold = 0 , repeat = 0 ):
5050 # Replace with retrieve from cache
Original file line number Diff line number Diff line change @@ -59,7 +59,7 @@ def test_get_cached_dataset_description_not_cached(self):
5959
6060 def test_get_cached_dataset_arff (self ):
6161 openml .config .set_cache_directory (self .static_cache_dir )
62- description = openml .datasets .functions ._get_cached_dataset_arff (2 )
62+ description = openml .datasets .functions ._get_cached_dataset_arff (did = 2 )
6363 self .assertIsInstance (description , str )
6464
6565 def test_get_cached_dataset_arff_not_cached (self ):
Original file line number Diff line number Diff line change @@ -32,7 +32,7 @@ def test_get_dataset(self, patch):
3232 def test_get_X_and_Y (self ):
3333 # Classification task
3434 task = openml .tasks .get_task (1 )
35- X , Y = task .get_X_and_Y ()
35+ X , Y = task .get_X_and_y ()
3636 self .assertEqual ((898 , 38 ), X .shape )
3737 self .assertIsInstance (X , np .ndarray )
3838 self .assertEqual ((898 , ), Y .shape )
@@ -41,7 +41,7 @@ def test_get_X_and_Y(self):
4141
4242 # Regression task
4343 task = openml .tasks .get_task (2280 )
44- X , Y = task .get_X_and_Y ()
44+ X , Y = task .get_X_and_y ()
4545 self .assertEqual ((8192 , 8 ), X .shape )
4646 self .assertIsInstance (X , np .ndarray )
4747 self .assertEqual ((8192 ,), Y .shape )
You can’t perform that action at this time.
0 commit comments