-
Notifications
You must be signed in to change notification settings - Fork 6
/
train_CNN.yaml
34 lines (29 loc) · 992 Bytes
/
train_CNN.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
model:
##################################################
# Pre-trained transfer model (typically rodent)
#
# Path
pre_trained_model_path: ""
#
# Filename of .h5 file (if empty string no pre-training will be done)
# pre_trained_model_file_name: "Liver_ballooning_model_V1.06.h5"
pre_trained_model_file_name: ""
##################################################
# Resulting model
#
# Path
model_path: "./model/"
#
# Filename of resulting model
model_file_name: "HS_Liver_ballooning_model_IN_V0.41_test.h5"
tiles:
# Path where the tiles with the ground truth are located in a folder ./train
ground_truth_path: "./CNN/fibrosis/"
settings:
# Max number of epochs to train
max_epochs: 30
# Generate a confusion matrix and save it
generate_confusion_matrix: True
# Automatically split train and validation data
do_val_split: False
val_fraction: 0.05