예제 #1
0
# -----------------------------------------------------------------------------
th = Config(as_global=True)
th.data_dir = from_root("FILOB/data/")
th.job_dir = from_root("FILOB")
# -----------------------------------------------------------------------------
# Device configurations
# -----------------------------------------------------------------------------
th.allow_growth = False
th.gpu_memory_fraction = 0.3

# -----------------------------------------------------------------------------
# Set information about the data set
# -----------------------------------------------------------------------------
th.max_level = 10
th.volume_only = True
th.developer_code = "use_log"

th.input_shape = [20]
th.num_classes = 3
th.output_dim = 3
th.target_dim = 1
th.target_dtype = tf.int32

th.val_size = 1

th.loss_string = "cross_entropy"

# -----------------------------------------------------------------------------
# Set common trainer configs
# -----------------------------------------------------------------------------
th.epoch = 1000  # STOCK 1000
예제 #2
0
# -----------------------------------------------------------------------------
th = Config(as_global=True)
th.data_dir = from_root('95-FI2010/data/')
th.job_dir = from_root('95-FI2010')
# -----------------------------------------------------------------------------
# Device configurations
# -----------------------------------------------------------------------------
th.allow_growth = False
th.gpu_memory_fraction = 0.3

# -----------------------------------------------------------------------------
# Set information about the data set
# -----------------------------------------------------------------------------
th.max_level = 10
th.volume_only = True
th.developer_code = 'use_log'

th.input_shape = [20]
th.num_classes = 3
th.output_dim = 3
th.target_dim = 1
th.target_dtype = tf.int32

th.val_size = 1

th.loss_string = 'cross_entropy'

# -----------------------------------------------------------------------------
# Set common trainer configs
# -----------------------------------------------------------------------------
th.epoch = 1000