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

Missing scikit-learn dependency for PD1 benchmark #19

Open
Bronzila opened this issue Jun 5, 2024 · 0 comments
Open

Missing scikit-learn dependency for PD1 benchmark #19

Bronzila opened this issue Jun 5, 2024 · 0 comments

Comments

@Bronzila
Copy link

Bronzila commented Jun 5, 2024

When I try to run the PD1 benchmark cifar100_wideresnet_2048, I get the error that sklearn is required for xgboost. After installing scikit-learn manually, everything works fine. It is simply missing in the dependencies or I messed something up during installation.

Steps to reproduce/How I installed MFPBench:

  1. Setup new conda env: conda create --name mfp_test python=3.8
  2. Activate environment: conda activate mfp_test
  3. Install mf-prior-bench: pip install mf-prior-bench[pd1]

Now when querying the benchmark it returns the error ImportError: sklearn needs to be installed in order to use this module.

After manually installing via pip install scikit-learn everything works fine.

Please let me know if I messed something up when setting up mfpbench :)

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

No branches or pull requests

1 participant