Bladeren bron

Refactor linear_regression for TF1.0

Signed-off-by: Norman Heckscher <norman.heckscher@gmail.com>
Norman Heckscher 8 jaren geleden
bovenliggende
commit
b557533091
2 gewijzigde bestanden met toevoegingen van 42 en 32 verwijderingen
  1. 1 1
      examples/2_BasicModels/linear_regression.py
  2. 41 31
      notebooks/2_BasicModels/linear_regression.ipynb

+ 1 - 1
examples/2_BasicModels/linear_regression.py

@@ -41,7 +41,7 @@ cost = tf.reduce_sum(tf.pow(pred-Y, 2))/(2*n_samples)
 optimizer = tf.train.GradientDescentOptimizer(learning_rate).minimize(cost)
 
 # Initializing the variables
-init = tf.initialize_all_variables()
+init = tf.global_variables_initializer()
 
 # Launch the graph
 with tf.Session() as sess:

File diff suppressed because it is too large
+ 41 - 31
notebooks/2_BasicModels/linear_regression.ipynb