Skip to content

Commit 935edca

Browse files
committed
adjusting docstring & default standardisation
1 parent f0a39bc commit 935edca

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

  • src/graphnet/models/data_representation/graphs/nodes

src/graphnet/models/data_representation/graphs/nodes/nodes.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -514,7 +514,7 @@ def __init__(
514514
time_spread: bool = True,
515515
time_std: bool = True,
516516
time_after_charge_pct: List[int] = [1, 3, 5, 11, 15, 20, 50, 80],
517-
charge_standardization: Union[float, str] = 1e-2,
517+
charge_standardization: Union[float, str] = "log",
518518
time_standardization: float = 1e-3,
519519
order_in_time: bool = True,
520520
add_counts: bool = False,
@@ -535,8 +535,9 @@ def __init__(
535535
time_std: If True, time std is added as a feature.
536536
time_after_charge_pct: List of percentiles to calculate time after
537537
charge.
538-
charge_standardization: Standardization factor for features
539-
with a charge value.
538+
charge_standardization: Either a float or 'log'. If a float,
539+
the features are multiplied by this factor. If 'log', the
540+
features are transformed to log10 scale.
540541
time_standardization: Standardization factor for features
541542
with a time
542543
order_in_time: If True, clusters are ordered in time.

0 commit comments

Comments
 (0)