-
Notifications
You must be signed in to change notification settings - Fork 368
/
run_ProtoNNTrain_usps10.sh
75 lines (51 loc) · 2.07 KB
/
run_ProtoNNTrain_usps10.sh
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT license.
# Have a look at README.md and README_PROTONN_OSS.md for details on how to setup the data directory to run this script.
########################################################
# Input-output parameters
########################################################
predefined_model="-P 0"
problem_format="-C 1"
train_file="-I usps10/train.txt"
validation_file="-V usps10/test.txt"
output_dir="-O usps10/ProtoNNResults"
#model_dir="-M usps10/"
input_format="-F 0"
########################################################
# Data-dependent parameters
########################################################
ntrain="-r 7291"
nvalidation="-v 2007"
original_dimension="-D 256"
num_labels="-l 10"
########################################################
# ProtoNN hyper-parameters (required)
########################################################
projection_dimension="-d 15"
#prototypes="-k 20"
prototypes="-m 200"
########################################################
# ProtoNN hyper-parameters (optional)
########################################################
lambda_W="-W 1.0"
lambda_Z="-Z 1.0"
lambda_B="-B 1.0"
gammaNumerator="-g 1.0"
normalization="-N 1"
seed="-R 42"
########################################################
# ProtoNN optimization hyper-parameters (optional)
########################################################
batch_size="-b 1024"
iters="-T 20"
epochs="-E 20"
########################################################
# execute ProtoNN
########################################################
#gdb=" gdb --args"
executable="./ProtoNNTrain"
command=$gdb" "$executable" "$predefined_model" "$seed" "$problem_format" "$train_file" "$validation_file" "$output_dir" "$model_dir" "$input_format" "$ntrain" "$nvalidation" "$original_dimension" "$projection_dimension" "$num_labels" "$prototypes" "$lambda_W" "$lambda_Z" "$lambda_B" "$gammaNumerator" "$batch_size" "$iters" "$epochs" "$normalization
echo "Running ProtoNN with following command: "
echo $command
echo ""
exec $command