def __init__(self, log_path, model_path): """ The PredictionEngine wraps the pair of generators/discriminators for both layers and configures the batch window Parameters: - window :: The batch window - log_path :: The directory path to which the logs will be saved model_path :: The directory path to which the models will be saved """ self.window_size = False self.default_model_save_path = "/home/dan/vestra/models/" self.log_path = log_path self.log_file = log_file self.db = DBObject() self.logger = ModelLogger(self.log_path, self.log_file) # check for cuda self.cuda = torch.cuda.is_available() self.learning_rate = 0.0002