Python truncated_normal_initializerの例

コード例 #1

ファイル: decisions_to_data.py プロジェクト: 1000sprites/tensorflow

  def _define_vars(self, params, **kwargs):
    with ops.device(self.device_assigner):

      self.tree_parameters = variable_scope.get_variable(
          name='hard_tree_parameters_%d' % self.layer_num,
          shape=[params.num_nodes, params.num_features],
          initializer=variable_scope.truncated_normal_initializer(
              mean=params.weight_init_mean, stddev=params.weight_init_std))

      self.tree_thresholds = variable_scope.get_variable(
          name='hard_tree_thresholds_%d' % self.layer_num,
          shape=[params.num_nodes],
          initializer=variable_scope.truncated_normal_initializer(
              mean=params.weight_init_mean, stddev=params.weight_init_std))

コード例 #2

    def _define_vars(self, params, **kwargs):
        with ops.device(self.device_assigner.get_device(self.layer_num)):

            self.tree_parameters = variable_scope.get_variable(
                name='hard_tree_parameters_%d' % self.layer_num,
                shape=[params.num_nodes, params.num_features],
                initializer=variable_scope.truncated_normal_initializer(
                    mean=params.weight_init_mean,
                    stddev=params.weight_init_std))

            self.tree_thresholds = variable_scope.get_variable(
                name='hard_tree_thresholds_%d' % self.layer_num,
                shape=[params.num_nodes],
                initializer=variable_scope.truncated_normal_initializer(
                    mean=params.weight_init_mean,
                    stddev=params.weight_init_std))