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

Torch #1

Open
wants to merge 81 commits into
base: main
Choose a base branch
from
Open

Torch #1

wants to merge 81 commits into from

Conversation

stroblme
Copy link
Member

  • added and manually optimized pytorch autoencoder (tensorflow AE can be launched by triggering dedicated pipeline)
  • splitted single pipeline into preprocessing, training and postprocessing
  • added poetry
  • added ssim metric replacing the pixel-wise accuracy
  • switched from matplotlib to plotly framework

pyproject.toml Outdated Show resolved Hide resolved
Copy link
Member

@eileen-kuehn eileen-kuehn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

There are some smaller comments and suggestions. When those are done, we are good to go 👍

pyproject.toml Outdated Show resolved Hide resolved
src/qc_data_preparation/helpers/dataset.py Show resolved Hide resolved
src/qc_data_preparation/helpers/dataset.py Outdated Show resolved Hide resolved
src/qc_data_preparation/helpers/types.py Outdated Show resolved Hide resolved
pyproject.toml Outdated Show resolved Hide resolved
eileen-kuehn and others added 4 commits June 25, 2023 23:34
…els (#7)

* feat: add model class to handle torch and tensorflow

* build: specify version for tensorflow

* feat: introduce number_of_features globally

---------

Co-authored-by: Max Fischer <[email protected]>
@stroblme stroblme requested a review from eileen-kuehn July 3, 2023 14:38
@stroblme stroblme mentioned this pull request Aug 10, 2023
2 tasks
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.

3 participants