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. |