Browse Source

Refactor linear_regression for TF1.0

Signed-off-by: Norman Heckscher <norman.heckscher@gmail.com>
Norman Heckscher 8 years ago
parent
commit
b557533091

+ 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