nn.py 文件源码

def __init__(self, dropout=0.2, cuda=False, cuda_device_id=0, standardize_observes=False, softmax_boost=20, mixture_components=10):
        super(Artifact, self).__init__()

        self.sample_layers = {}
        self.proposal_layers = {}
        self.observe_layer = None
        self.lstm = None

        self.model_name = ''
        self.created = util.get_time_str()
        self.modified = util.get_time_str()
        self.on_cuda = cuda
        self.trained_on = ''
        self.cuda_device_id = cuda_device_id
        self.code_version = pyprob.__version__
        self.pytorch_version = torch.__version__
        self.standardize_observes = standardize_observes
        self.one_hot_address = {}
        self.one_hot_distribution = {}
        self.one_hot_address_dim = None
        self.one_hot_distribution_dim = None
        self.one_hot_address_empty = None
        self.one_hot_distribution_empty = None
        self.address_histogram = {}
        self.trace_length_histogram = {}
        self.valid_size = None
        self.valid_batch = None
        self.lstm_dim = None
        self.lstm_depth = None
        self.lstm_input_dim = None
        self.smp_emb_dim = None
        self.obs_emb = None
        self.obs_emb_dim = None
        self.num_params_history_trace = []
        self.num_params_history_num_params = []
        self.trace_length_min = sys.maxsize
        self.trace_length_max = 0
        self.trace_examples_histogram = {}
        self.trace_examples_addresses = {}
        self.trace_examples_limit = 10000
        self.train_loss_best = None
        self.train_loss_worst = None
        self.valid_loss_best = None
        self.valid_loss_worst = None
        self.valid_history_trace = []
        self.valid_history_loss = []
        self.train_history_trace = []
        self.train_history_loss = []
        self.total_training_seconds = 0
        self.total_iterations = 0
        self.total_traces = 0
        self.updates = 0
        self.optimizer = None
        self.dropout = dropout
        self.softmax_boost = softmax_boost
        self.mixture_components = mixture_components

        self._state_observes = None
        self._state_observes_embedding = None
        self._state_new_trace = True