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

Adding NN and CNN class implementations using Shadow #50

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

Conversation

stompsjo
Copy link
Collaborator

I have pushed my stompsjo/RadClass:shadow branch to cnerg/RadClass:shadow. I will close #44 and copy its contents here for review.

This introduces two classes (ShadowNN and ShadowCNN) that can be used for semi-supervised machine learning with neural networks, Shadow, and PyTorch. This includes typical scikit-learn-esque methods like train and predict as well as methods for hyperparameter optimization and saving the model to file.

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