File size: 532 Bytes
5fa1a76 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
thon
from transformers import PatchTSMixerConfig, PatchTSMixerForPrediction
from transformers import Trainer, TrainingArguments,
config = PatchTSMixerConfig(context_length = 512, prediction_length = 96)
model = PatchTSMixerForPrediction(config)
trainer = Trainer(model=model, args=training_args,
train_dataset=train_dataset,
eval_dataset=valid_dataset)
trainer.train()
results = trainer.evaluate(test_dataset)
Usage tips
The model can also be used for time series classification and time series regression. |