Victoria BOURGEAIS

provide supplementary functions to the main scripts

modify the .gitignore
...@@ -3,3 +3,4 @@ first_matrix_connection_GO.csv ...@@ -3,3 +3,4 @@ first_matrix_connection_GO.csv
3 go_level.csv 3 go_level.csv
4 X_test.npz 4 X_test.npz
5 X_train.npz 5 X_train.npz
6 +__pycache__/
......
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()),
......
This diff is collapsed. Click to expand it.