search

rajatsen91 / MFTreeSearchCV

Hyper-parameter Tuning through noisy multi-fidelity tree search
MIT License
8 stars 1 forks source link

Experiments about the real work #1

Open jxiw opened 3 years ago

jxiw commented 3 years ago

It seems this repo only contains Synthetic Experiments. Can we also access the code related to real experiments, e.g., XGB on MNIST, SVM and ConvNet? Thanks a lot.

rajatsen91 commented 3 years ago

The code is generalizable to any sklearn format functions as shown in the example given in the README: https://github.com/rajatsen91/MFTreeSearchCV/blob/master/README.md#example

For the Convnet experiment we have to write the training and prediction as an sklearn estimator with a fit and a predict method.