-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathconstants_insertion.py
More file actions
50 lines (35 loc) · 1.66 KB
/
constants_insertion.py
File metadata and controls
50 lines (35 loc) · 1.66 KB
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
43
44
45
46
47
48
49
50
# Author : Debanjali Biswas
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Constants and File Paths
"""
# Model Constants
HIDDEN_DIM1 = 128 # Hidden dimension 1
HIDDEN_DIM2 = 32 # Hidden dimension 2
OUTPUT_DIM = 1 # Classifier output_dim
LR = 0.0001 # Learning rate for optimizer
MAX_EPOCHS = 70 # Training Epochs
CUDA_DEVICE = "cuda:0" # GPU
#####################################
# Model file paths and names
destination_folder1 = "./results1_insertions" # Destination folder for the trained model and metrics for extended model
destination_folder2 = "./results2" # Destination folder for the trained model and metrics for base model
destination_folder3 = "./results3" # Destination folder for the trained model and metrics for cosine similarity baseline model
destination_folder4 = "./results4" # Destination folder for the heuristics for naive baseline model
prediction_file = "insertion.tsv" # Testing results
#####################################
# Data file paths and names
folder = "./data" # Dataset folder
test_folder = "./test" # Folder with test data
recipe_folder_name = "recipes" # Folder containing recipes
alignment_file = "alignments.tsv" # Alignment file