PatchTSTModel, PatchTSTConfig, & Trainer

[Datakunskap]

Feature request

Implementation of:
PatchTSTModel, PatchTSTConfig, & Trainer

Source:
https://github.com/huggingface/transformers/blob/v4.46.3/src/transformers/models/patchtst/modeling_patchtst.py#L1142

Motivation

https://github.com/huggingface/transformers

Your contribution

Submitting a PR

[xenova]

Opened PR here. Let me know if that works for you 👍

[Datakunskap]

@xenova You're amazing! We were about to switch over to Python. You saved our Typescript devs. Thank you!

We have one more request to make it on par with the Python transformer library. Would it be possible to add the following Models as well?

• PatchTSMixer
• PatchTSMixerForPrediction
• PatchTSMixerConfig

Kindest regards,
Datakunskap

[xenova]

Sure thing! I've added them here.

Example usage for PatchTSMixerForPrediction can be found here.

Note for the config, you an simply do:

import { AutoConfig } from '@huggingface/transformers';

const config = await AutoConfig.from_pretrained('hf-internal-testing/tiny-random-PatchTSTModel'); 
console.log(config);