diff --git a/MetaAugment/README.md b/MetaAugment/README.md index 18c2142165665ea15fa85fa4009e40ea307f091f..448ee7cef92f512c2596b2b51b4dfc280fb25676 100644 --- a/MetaAugment/README.md +++ b/MetaAugment/README.md @@ -1,4 +1,4 @@ -# MetaRL +# AutoAugment Library Documentation: https://metaaugment.readthedocs.io/en/latest/ \ No newline at end of file diff --git a/MetaAugment/agent_gen.py b/MetaAugment/agent_gen.py deleted file mode 100644 index 15653a64292cc6863ffe65f033d64ca02d467188..0000000000000000000000000000000000000000 --- a/MetaAugment/agent_gen.py +++ /dev/null @@ -1,25 +0,0 @@ -from MetaAugment.autoaugment_learners.aa_learner import augmentation_space, aa_learner -from MetaAugment.autoaugment_learners.evo_learner import evo_learner -from MetaAugment.autoaugment_learners.gru_learner import gru_learner -from MetaAugment.autoaugment_learners.randomsearch_learner import randomsearch_learner - -def gen_learner(name: str, **kwags): - """ - Generates a learner based on input from the user. - - Parameters - ------------- - name -> string - Of form 'gru', 'rand', 'evo' for GRU, Random search, or Evolutionary - learner generation - - **kwags -> key word arguments for respective learner - """ - name == name.lower() - if name == "gru": - agent = gru_learner(**kwags) - elif name == "evo": - agent = evo_learner(**kwags) - elif name == "rand" or name == "random" or name == "ran": - agent = randomsearch_learner(**kwags) - return agent diff --git a/MetaAugment/saved_models/mnist_0.9823.pkl b/MetaAugment/saved_models/mnist_0.9823.pkl deleted file mode 100644 index c762f2750da9807132221b447aef89c0211b36bd..0000000000000000000000000000000000000000 Binary files a/MetaAugment/saved_models/mnist_0.9823.pkl and /dev/null differ