diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/debug_test.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/debug_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/debug_test.py b/tensorflow/contrib/learn/python/learn/estimators/debug_test.py index b4b552c025..6b125534a4 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/debug_test.py +++ b/tensorflow/contrib/learn/python/learn/estimators/debug_test.py @@ -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 |