How to train?

[wtt6668888]

How should the provided code be applied to your own dataset for training and prediction?

[fayvor]

Hi @wtt6668888, here are a couple of notebooks that may be of use to you. Your dataset will need to be parseable into a pandas dataframe with a time column and a target column for prediction. Please let me know if this helps answer your question.
Energy Demand Forecasting - Basic Inference
Preprocessing and Performance Evaluation

[wgifford]

@wtt6668888 For pretraining a new model from scratch you can take a look at the workflow here: https://github.com/ibm-granite/granite-tsfm/blob/main/notebooks/hfdemo/tinytimemixer/ttm_pretrain_sample.py

If you wish to fine-tune the pretrained model, you can take a look at the following notebook: https://github.com/ibm-granite/granite-tsfm/blob/main/notebooks/tutorial/ttm_with_exog_tutorial.ipynb

This notebook fine-tunes the model so that additional exogenous features can be included.