Skip to content

Commit f81d037

Browse files
committed
update
1 parent 9c28457 commit f81d037

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tensorflowTUT/tf17_dropout/full_code.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from __future__ import print_function
1010
import tensorflow as tf
1111
from sklearn.datasets import load_digits
12-
from sklearn.cross_validation import train_test_split
12+
from sklearn.model_selection import train_test_split
1313
from sklearn.preprocessing import LabelBinarizer
1414

1515
# load data
@@ -63,7 +63,6 @@ def add_layer(inputs, in_size, out_size, layer_name, activation_function=None, )
6363
else:
6464
init = tf.global_variables_initializer()
6565
sess.run(init)
66-
6766
for i in range(500):
6867
# here to determine the keeping probability
6968
sess.run(train_step, feed_dict={xs: X_train, ys: y_train, keep_prob: 0.5})

0 commit comments

Comments
 (0)