Skip to content

Commit 4c458db

Browse files
committed
add default preprocessing
1 parent 920ec01 commit 4c458db

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

torch_geometric/datasets/graphland.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)