search

LMZimmer / Auto-PyTorch_refactor

Apache License 2.0
0 stars 2 forks source link

Image Pipeline #14

Closed ravinkohli closed 3 years ago

ravinkohli commented 3 years ago
  1. Split preprocessing pipeline into tabular and image preprocessing
  2. added preprocessing node
  3. added Image classification pipeline with example on fashion-mnist
franchuterivera commented 3 years ago

@ravinkohli Do you think you can make sure examples/examples_tabular_classification.py is able to run?

I think right now it is broken because torchvision is not added in the dependencies

(and if you fix that dependency, other things are still pending to make sure this file doesn't crash)

franchuterivera commented 3 years ago

some mypy test was still failing. Also the example for tabular classification was not working, but all minor changes. So I added the changes myself to expedite the process. All comments have been addressed so I am merging this now!