@@ -29,8 +29,8 @@ def fit(self, y):
2929 """
3030 Fit label encoder to a given set of labels.
3131
32- Parameters:
33- -----------
32+ Parameters
33+ ----------
3434 y : array-like
3535 Labels to encode.
3636 """
@@ -42,13 +42,13 @@ def transform(self, y):
4242 """
4343 Transform labels to encoded integer labels.
4444
45- Parameters:
46- -----------
45+ Parameters
46+ ----------
4747 y : array-like (1-D vector)
4848 Labels to encode.
4949
50- Returns:
51- --------
50+ Returns
51+ -------
5252 encoded_labels : array-like
5353 Encoded integer labels.
5454 """
@@ -77,13 +77,13 @@ def inverse_transform(self, y):
7777 """
7878 Transform integer labels to original labels.
7979
80- Parameters:
81- -----------
80+ Parameters
81+ ----------
8282 y : array-like
8383 Encoded integer labels.
8484
85- Returns:
86- --------
85+ Returns
86+ -------
8787 original_labels : array-like
8888 Original labels.
8989 """
@@ -107,7 +107,8 @@ def difference(self, X):
107107 def inverse_difference (self , diff_data ):
108108 if self .original_data is None :
109109 raise ValueError ("Original data is required for inversion." )
110- return np .array ([diff_data [i - self .interval ] + self .original_data [i - self .interval ] for i in range (self .interval , len (self .original_data ))])
110+ return np .array ([diff_data [i - self .interval ] + self .original_data [i - self .interval ] for i in
111+ range (self .interval , len (self .original_data ))])
111112
112113
113114class FeatureEngineering :
@@ -183,7 +184,7 @@ def check_y(y):
183184 return y
184185
185186 @staticmethod
186- def scale (X , scaling_methods = ('standard' , ), list_dict_paras = None ):
187+ def scale (X , scaling_methods = ('standard' ,), list_dict_paras = None ):
187188 X = np .squeeze (np .asarray (X ))
188189 if X .ndim == 1 :
189190 X = np .reshape (X , (- 1 , 1 ))
@@ -208,7 +209,9 @@ def split_train_test(self, test_size=0.2, train_size=None,
208209 The wrapper of the split_train_test function in scikit-learn library.
209210 """
210211 self .X_train , self .X_test , self .y_train , self .y_test = train_test_split (self .X , self .y , test_size = test_size ,
211- train_size = train_size , random_state = random_state , shuffle = shuffle , stratify = stratify )
212+ train_size = train_size ,
213+ random_state = random_state ,
214+ shuffle = shuffle , stratify = stratify )
212215 if not inplace :
213216 return self .X_train , self .X_test , self .y_train , self .y_test
214217
0 commit comments