Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TemporalFusionTransformer truss example #367

Merged
merged 5 commits into from
Jun 14, 2023
Merged

Conversation

spal1
Copy link
Collaborator

@spal1 spal1 commented Jun 14, 2023

This PR adds a truss for the TemporalFusionTransformer model following this tutorial from pytorch-forecasting.

To invoke the truss locally, use the following, where new_prediction_data is taken from the pytorch-forecasting example here

import truss
tft = truss.load('temporal-fusion-transformer')
preds = tft.docker_predict({"data": new_prediction_data.to_json(orient="records")})

@spal1 spal1 requested review from bolasim and joostinyi June 14, 2023 18:18
@bolasim bolasim merged commit 5025546 into main Jun 14, 2023
2 checks passed
@bolasim bolasim deleted the samiksha/add-tft-truss branch June 14, 2023 18:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants