automl / NASLib

NASLib is a Neural Architecture Search (NAS) library for facilitating NAS research for the community by providing interfaces to several state-of-the-art NAS search spaces and optimizers.
Apache License 2.0
512 stars 117 forks source link

Adding ninapro and darcyflow datasets #163

Closed Rufaim closed 1 year ago

Rufaim commented 1 year ago

Hi, Here is a couple of datasets from from NAS-Bench-360. While i added dataloaders for both only ninapro can be incorporated into NAS-Bench-201 without any substantial api changes. Archive with NinaPro evaluations for can be downloaded from here. To download data use this link.

Rufaim commented 1 year ago

I added a small script that allows to generate custom dataset splits for Taskonomy data. Now repository contain all the scripts to download and run "autoencoder", "class_object", "class_scene", "normal" and "jigsaw" tasks.

Neonkraft commented 1 year ago

Hi @Rufaim,

Thank you very much for this PR! We've taken a look at it, and it looks good to us. @shakibamrd is currently looking into the one test that's failing - it seems to stem from the version of scipy that is being used. Once we know which version works best, I'll be glad to merge this PR!

Best, Arjun