def __init__(self, api_url, mongo_uri, **kwargs): self.url = api_url self.name = socket.gethostname() self.http_client = AsyncHTTPClient() self.models = {"keras": KerasModel(**kwargs)} self.data_path = os.path.join(os.path.dirname(__file__), "data") self.data_loader = DataLoader(api_url, mongo_uri)
def __init__(self, exp_id, nb_epoch = 10, patience = 5, mongo_uri = "mongodb://localhost:27017/", data_filename = None): self.exp_id = exp_id self.model = None self.data = None self.data_loader = None # How many epoch to train (if score earlystopping doesn't kick in earlier) self.nb_epoch = nb_epoch # How many epoch of not improving before earlystopper kicks in self.patience = patience self.mongo_uri = mongo_uri self.factory = KerasModelFactory() self.data_loader = DataLoader("http://localhost:8080", self.mongo_uri)