-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yaml
42 lines (32 loc) · 1.82 KB
/
config.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
35
36
37
38
39
40
41
42
experiment_name: 'test'
model: 'dg1' # Available choices: dg1
mode: 'ti' # pretrain, train, inference, pt, pti, ti
cas: 'cas9_nucleosome' # cas9_seq, cas9_nucleosome
guide_length: 28 # Default 28
# Only if pretraining. fasta file containing the entire genome.
# Place in the input directory.
pretrain_genome_input_name: 'CBS6556GenomewSimplifiedHeaders'
train_guides_csv_file_name: 'example_train.csv' # Must be a .csv
train_guide_seq_col_name: 'guide_with_context' # The name of the column in the training set that contains the guides with flanking context base pairs
train_guide_score_col_name: '30dC_2%_glucose_CS_day2_normalized_by_NT_avg' # The name of the column in the training set that contains the true experimental scores
train_nucleosome_occupancy_col_name: 'nucleosome_occupancy'
inference_guides_csv_file_name: 'example_test.csv'
inference_guide_seq_col_name: 'guide_with_context'
inference_nucleosome_occupancy_col_name: 'nucleosome_occupancy'
# Output a diagram that shows the ML architecture?
# Requires Pydot, and Pydotplus -- Not required to run an experiment.
# Feel free to disable (set to False) if you don't have those packages.
plot_model: False
# ----------------------
metrics: False # Switch to True for large enough data -- Won't work with examples
plot_auc_curve: False # Switch to True for large enough data -- Won't work with examples
# ----------------------
# DG1 HYPERPARAMETERS
dg_one_pretrain_train_test_ratio: 0.7 # Default: 0.7 Means 70% of data used for training and 30% for validation.
dg_one_train_test_ratio: 0.7 # Default: 0.7
dg_one_adam_lr: 0.003 # Default 0.001 - Used for both pretrain and train models
dg_one_roc_curve_threshold: 3 # Default 3
dg_one_pretrain_batch_size: 64 # Default 64
dg_one_batch_size: 64 # Default 64
dg_one_pretrain_epochs: 6 # Default 6
dg_one_epochs: 10 # Default 10