Skip to content
Snippets Groups Projects
Commit 489e7f79 authored by Sun Jin Kim's avatar Sun Jin Kim
Browse files

delete obsolete files in MetaAugment

parent 0ac93b69
No related branches found
No related tags found
No related merge requests found
Pipeline #272012 passed
# MetaRL
# AutoAugment Library
Documentation:
https://metaaugment.readthedocs.io/en/latest/
\ No newline at end of file
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
File deleted
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment