Skip to content

David work vivit#2

Open
hwalters361 wants to merge 11 commits intodavid-workfrom
david-work-vivit
Open

David work vivit#2
hwalters361 wants to merge 11 commits intodavid-workfrom
david-work-vivit

Conversation

@hwalters361
Copy link
Copy Markdown
Owner

I added a cell to process ppt and sst data in a way which preserves the shape of the data . It does not remove 'nan' entries, so that the shape of the data is (732, 24, 89, 180) and not (732, 263712). This is helpful because we want the ViViT transformer model to process the data like a video with 24 frames. This reformatted data is then saved under the directory "data/sst.npz" in a later code cell in order for the ViViT pipeline to work properly.

I also added a file called vivit.py which contains the code for the ViViT transformer model. The Introduction.ipynb takes this code and runs a model for 15 epochs over the labels for cluster 1.

None of this code should interfere with the code cells of previous work, or PCA work.

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.

2 participants