Unverified Commit 6ff10922 authored by vijaye12's avatar vijaye12 Committed by GitHub
Browse files

Fix PatchTSMixer Docstrings (#27943)



* docstring corrections

* style make

---------
Co-authored-by: default avatarvijaye12 <vijaye12@in.ibm.com>
parent accccdd0
......@@ -49,8 +49,6 @@ class PatchTSMixerConfig(PretrainedConfig):
non-overlapping patches.
num_parallel_samples (`int`, *optional*, defaults to 100):
The number of samples to generate in parallel for probabilistic forecast.
d_model (`int`, *optional*, defaults to 8):
Hidden dimension of the model. Recommended to set it as a multiple of patch_length (i.e. 2-5X of
patch_len). Larger value indicates more complex model.
......@@ -96,8 +94,6 @@ class PatchTSMixerConfig(PretrainedConfig):
`PyTorch`. Setting it to `False` performs `PyTorch` weight initialization.
norm_eps (`float`, *optional*, defaults to 1e-05):
A value added to the denominator for numerical stability of normalization.
mask_type (`str`, *optional*, defaults to `"random"`):
Type of masking to use for Masked Pretraining mode. Allowed values are "random", "forecast". In Random
masking, points are masked randomly. In Forecast masking, points are masked towards the end.
......@@ -116,9 +112,6 @@ class PatchTSMixerConfig(PretrainedConfig):
across channels.
unmasked_channel_indices (`list`, *optional*):
Channels that are not masked during pretraining.
head_dropout (`float`, *optional*, defaults to 0.2):
The dropout probability the `PatchTSMixer` head.
distribution_output (`string`, *optional*, defaults to `"student_t"`):
......
......@@ -1665,7 +1665,6 @@ class PatchTSMixerForPrediction(PatchTSMixerPreTrainedModel):
to forecast only specific channels by setting the indices in `prediction_channel_indices` parameter,
pass the target data with all channels, as channel Filtering for both prediction and target will be
manually applied before the loss computation.
return_loss (`bool`, *optional*):
Whether to return the loss in the `forward` call.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment