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

LinTzuYaunDataset datasets, Classification problem support added to HGNN #7

Merged
merged 17 commits into from
Jul 3, 2024

Conversation

DanielChaseButterfield
Copy link
Member

LinTzuYaunDataset datasets, Classification problem support added to HGNN

Features

  • LinTzuYaun Datasets implemented, which use foot positions and velocities with class labels instead of GRF labels.
  • Huge Dataset code refactor which allows all datasets to reuse the same get() functions and sorting functions. Even datasets with different inputs can reuse these same functions, as the functions will check the returned dataset arrays and omit any that are None.
  • Classification model support added to HGNN models, using Cross-Entropy loss for training and Accuracy for evaluation.
  • Mini Cheetah URDF model added.
  • Sweeper automatically generates a visualization of the model performance (for regression models).
  • A very simple ANN dataset class implemented with limited functionality.

Organization

  • datasets.py file seperated into multiple python files in the "datasets_py" folder.

Bug Fixes

  • Fixed a bug where the model name saved didn't match the sweeper name.

… train_model, Test cases for loss calculations
… of FlexibleDataset so more functions can be reused (not ready yet)
… gnn, and heterogeneous gnn creation methods with all datasets, removed start and end sequence method
…(due to removal of derivative calculation), add functionality to check that a dataset path provided by the user matches what the dataset class expects
…ted to ensure it works whether or not foot pos/vel data is provided
@DanielChaseButterfield DanielChaseButterfield changed the title Develop LinTzuYaunDataset datasets, Classification problem support added to HGNN Jul 3, 2024
@DanielChaseButterfield DanielChaseButterfield merged commit bf62242 into main Jul 3, 2024
2 checks passed
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.

1 participant