provide supplementary functions to the main scripts
modify the .gitignore
Showing
4 changed files
with
9 additions
and
5 deletions
DeepGONet.py
0 → 100644
This diff is collapsed. Click to expand it.
... | @@ -6,13 +6,11 @@ FLAGS = tf.app.flags.FLAGS | ... | @@ -6,13 +6,11 @@ FLAGS = tf.app.flags.FLAGS |
6 | class BaseModel(): | 6 | class BaseModel(): |
7 | 7 | ||
8 | def __init__(self,X,n_input,n_classes,n_hidden_1,n_hidden_2,n_hidden_3,n_hidden_4,n_hidden_5,n_hidden_6,is_training,keep_prob): | 8 | def __init__(self,X,n_input,n_classes,n_hidden_1,n_hidden_2,n_hidden_3,n_hidden_4,n_hidden_5,n_hidden_6,is_training,keep_prob): |
9 | + | ||
10 | + | ||
11 | + # Parameters | ||
9 | self.X = X | 12 | self.X = X |
10 | self.n_input = n_input | 13 | self.n_input = n_input |
11 | - self.is_training = is_training | ||
12 | - | ||
13 | - #Hyperparameters | ||
14 | - self.keep_prob = keep_prob # Dropout | ||
15 | - | ||
16 | self.n_classes=n_classes | 14 | self.n_classes=n_classes |
17 | self.n_hidden_1=n_hidden_1 | 15 | self.n_hidden_1=n_hidden_1 |
18 | self.n_hidden_2=n_hidden_2 | 16 | self.n_hidden_2=n_hidden_2 |
... | @@ -21,6 +19,11 @@ class BaseModel(): | ... | @@ -21,6 +19,11 @@ class BaseModel(): |
21 | self.n_hidden_5=n_hidden_5 | 19 | self.n_hidden_5=n_hidden_5 |
22 | self.n_hidden_6=n_hidden_6 | 20 | self.n_hidden_6=n_hidden_6 |
23 | 21 | ||
22 | + # Hyperparameters | ||
23 | + self.keep_prob = keep_prob # Dropout | ||
24 | + self.is_training = is_training # BN | ||
25 | + | ||
26 | + | ||
24 | def store_layer_weights_and_bias(self): | 27 | def store_layer_weights_and_bias(self): |
25 | self.weights = { | 28 | self.weights = { |
26 | 'h1_w': tf.get_variable('W1', shape=(self.n_input, self.n_hidden_1), initializer=tf.contrib.layers.variance_scaling_initializer()), | 29 | 'h1_w': tf.get_variable('W1', shape=(self.n_input, self.n_hidden_1), initializer=tf.contrib.layers.variance_scaling_initializer()), | ... | ... |
train.py
deleted
100644 → 0
This diff is collapsed. Click to expand it.
-
Please register or login to post a comment