Skip to content

Instantly share code, notes, and snippets.

@oscar-defelice
Last active March 17, 2021 17:09
Show Gist options
  • Save oscar-defelice/e1d4501c24155cd357fae436554c39db to your computer and use it in GitHub Desktop.
Save oscar-defelice/e1d4501c24155cd357fae436554c39db to your computer and use it in GitHub Desktop.
{
"data": "data.json",
"train_test_ratio": 0.2,
"normalise": true,
"pretrained_model": null,
"model_config": {
"model_name": "house_pricing_model",
"layers": {
"first_layer": 64,
"second_layer": 64,
"third_layer": 32,
"output_layer": 1
},
"activations": ["relu", "relu", "relu", null],
"loss_function": "mse",
"optimiser": "adam",
"metrics": ["mae"]
},
"training_config": {
"batch_size": 10,
"epochs": 17
},
"save_model": true,
"model_path": "model/model.h5"
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment