experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make debug_test slice integer.

Browse Source
This commit is contained in:
Yifei Feng 2017-06-21 11:19:51 -07:00
parent f69c7569cd
commit a286b7db8e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/contrib/learn/python/learn/estimators/debug_test.py
View File

@ -52,8 +52,8 @@ LABEL_DIMENSION = 3 # Dimensionality of regression labels.
def _train_test_split(features_and_labels):
features, labels = features_and_labels
train_set = (features[:len(features) / 2], labels[:len(features) / 2])
test_set = (features[len(features) / 2:], labels[len(features) / 2:])
train_set = (features[:int(len(features) / 2)], labels[:int(len(features) / 2)])
test_set = (features[int(len(features) / 2):], labels[int(len(features) / 2):])
return train_set, test_set