示例#1
0
    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