@@ -211,11 +211,10 @@ def __init__(
211211 root : str ,
212212 name : str ,
213213 split : str ,
214- numerical_features_transform : Optional [
215- str ] = 'quantile_transform_normal' ,
216- fraction_features_transform : Optional [str ] = None ,
214+ numerical_features_transform : Optional [str ] = 'default' ,
215+ fraction_features_transform : Optional [str ] = 'default' ,
217216 categorical_features_transform : Optional [str ] = 'one_hot_encoding' ,
218- regression_targets_transform : Optional [str ] = 'standard_scaler ' ,
217+ regression_targets_transform : Optional [str ] = 'default ' ,
219218 numerical_features_nan_imputation_strategy : Optional [
220219 str ] = 'most_frequent' ,
221220 fraction_features_nan_imputation_strategy : Optional [
@@ -239,6 +238,17 @@ def __init__(
239238 ], ('Temporal split is not available for city-reviews, '
240239 'city-roads-M, city-roads-L, web-trafic.' )
241240
241+ if numerical_features_transform == 'default' :
242+ numerical_features_transform = 'quantile_transform_normal'
243+
244+ if fraction_features_transform == 'default' :
245+ fraction_features_transform = ( #
246+ 'quantile_transform_normal'
247+ if name in ['artnet-views' , 'avazu-ctr' ] else None )
248+
249+ if regression_targets_transform == 'default' :
250+ regression_targets_transform = 'standard_scaler'
251+
242252 if numerical_features_transform is not None :
243253 assert numerical_features_transform in [
244254 'standard_scaler' ,
0 commit comments