thereadme/world-ecoregion
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix(map-generator): improve continent generation

Browse Source
This commit is contained in:
Mahdi Dibaiee
2019-04-24 15:00:45 +04:30
parent 3bec4d7486
commit b377c6dd5f
3 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
biomes/train.py
View File

@ -46,9 +46,9 @@ class TuneB(tune.Trainable):
return self.model.restore(path)
A_params = {
'batch_size': tune.grid_search([5, 16, 32, 64]),
'layers': tune.grid_search([[16, 16], [32, 32], [128, 128]]),
'lr': tune.grid_search([1e-4, 1e-3, 1e-2]),
'batch_size': tune.grid_search([32]),
'layers': tune.grid_search([[32, 32]]),
'lr': tune.grid_search([1e-4]),
'optimizer': tune.grid_search([tf.keras.optimizers.Adam]),
}

2
biomes/utils.py
View File

@ -102,7 +102,7 @@ def dataframe_to_dataset_temp_precip(df):
tf_output = np.concatenate((tf_output, df[output].values), axis=0)
tf_inputs = tf.cast(normalize_ndarray(tf_inputs), tf.float32)
tf_output = tf.cast(tf_output, tf.float32)
tf_output = tf.cast(normalize_ndarray(tf_output), tf.float32)
logger.debug('dataset size: rows=%d, input_columns=%d, num_classes=%d', int(tf_inputs.shape[0]), input_columns, num_classes)
return int(tf_inputs.shape[0]), input_columns, num_classes, None, tf.data.Dataset.from_tensor_slices((tf_inputs, tf_output))