Update README.md
Browse files
README.md
CHANGED
|
@@ -114,11 +114,11 @@ num_samples = 100
|
|
| 114 |
# For inference_token_len, you can refer to LightGTS (Periodic Patching).
|
| 115 |
# We recommend to use the period length as the inference_token_len.
|
| 116 |
output = model.generate(inputs=batch_x,text_input_ids=batch_input_ids,
|
| 117 |
-
|
| 118 |
text_token_type_ids=batch_token_type_ids,
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
|
| 123 |
|
| 124 |
# use raw predictions for mean/quantiles/confidence-interval estimation
|
|
|
|
| 114 |
# For inference_token_len, you can refer to LightGTS (Periodic Patching).
|
| 115 |
# We recommend to use the period length as the inference_token_len.
|
| 116 |
output = model.generate(inputs=batch_x,text_input_ids=batch_input_ids,
|
| 117 |
+
text_attention_mask=batch_attention_mask,
|
| 118 |
text_token_type_ids=batch_token_type_ids,
|
| 119 |
+
max_output_length=forecast_length,
|
| 120 |
+
num_samples=num_samples,
|
| 121 |
+
inference_token_len=48)
|
| 122 |
|
| 123 |
|
| 124 |
# use raw predictions for mean/quantiles/confidence-interval estimation
|