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

lint errors

Browse Source
This commit is contained in:
nrstott 2018-07-02 12:37:15 -04:00
parent bd9b328012
commit f886f5377f
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/estimator/inputs/pandas_io.py
View File

@ -112,7 +112,7 @@ def pandas_input_fn(x,
'Index for y: %s\n' % (x.index, y.index)) 'Index for y: %s\n' % (x.index, y.index))
if isinstance(y, pd.DataFrame): if isinstance(y, pd.DataFrame):
y_columns = [(column, _get_unique_target_key(x, column)) y_columns = [(column, _get_unique_target_key(x, column))
for column in list(y)] for column in list(y)]
target_column = [v for _, v in y_columns] target_column = [v for _, v in y_columns]
x[target_column] = y x[target_column] = y
else: else:

16
tensorflow/python/estimator/inputs/pandas_io_test.py
View File

@ -84,9 +84,9 @@ class PandasIoTest(test.TestCase):
with self.assertRaisesRegexp(TypeError, with self.assertRaisesRegexp(TypeError,
'target_column must be a string type'): 'target_column must be a string type'):
pandas_io.pandas_input_fn(x, y, batch_size=2, pandas_io.pandas_input_fn(x, y, batch_size=2,
shuffle=False, shuffle=False,
num_epochs=1, num_epochs=1,
target_column=['one', 'two']) target_column=['one', 'two'])
def testPandasInputFn_NonBoolShuffle(self): def testPandasInputFn_NonBoolShuffle(self):
if not HAS_PANDAS: if not HAS_PANDAS:
@ -119,7 +119,7 @@ class PandasIoTest(test.TestCase):
with self.test_session() as session: with self.test_session() as session:
x, y = self.makeTestDataFrameWithYAsDataFrame() x, y = self.makeTestDataFrameWithYAsDataFrame()
input_fn = pandas_io.pandas_input_fn( input_fn = pandas_io.pandas_input_fn(
x, y, batch_size=2, shuffle=False, num_epochs=1) x, y, batch_size=2, shuffle=False, num_epochs=1)
features, targets = self.callInputFnOnce(input_fn, session) features, targets = self.callInputFnOnce(input_fn, session)
@ -128,14 +128,14 @@ class PandasIoTest(test.TestCase):
self.assertAllEqual(targets['a_target'], [10, 11]) self.assertAllEqual(targets['a_target'], [10, 11])
self.assertAllEqual(targets['b_target'], [50, 51]) self.assertAllEqual(targets['b_target'], [50, 51])
def testPandasInputFnWhenYIsDataFrame_HandlesOverlappingColumnNames(self): def testPandasInputFnYIsDataFrame_HandlesOverlappingColumns(self):
if not HAS_PANDAS: if not HAS_PANDAS:
return return
with self.test_session() as session: with self.test_session() as session:
x, y = self.makeTestDataFrameWithYAsDataFrame() x, y = self.makeTestDataFrameWithYAsDataFrame()
y = y.rename(columns={'a_target': 'a', 'b_target': 'b'}) y = y.rename(columns={'a_target': 'a', 'b_target': 'b'})
input_fn = pandas_io.pandas_input_fn( input_fn = pandas_io.pandas_input_fn(
x, y, batch_size=2, shuffle=False, num_epochs=1) x, y, batch_size=2, shuffle=False, num_epochs=1)
features, targets = self.callInputFnOnce(input_fn, session) features, targets = self.callInputFnOnce(input_fn, session)
@ -144,14 +144,14 @@ class PandasIoTest(test.TestCase):
self.assertAllEqual(targets['a'], [10, 11]) self.assertAllEqual(targets['a'], [10, 11])
self.assertAllEqual(targets['b'], [50, 51]) self.assertAllEqual(targets['b'], [50, 51])
def testPandasInputFnWhenYIsDataFrame_HandlesOverlappingColumnNamesInTargets(self): def testPandasInputFnYIsDataFrame_HandlesOverlappingColumnsInTargets(self):
if not HAS_PANDAS: if not HAS_PANDAS:
return return
with self.test_session() as session: with self.test_session() as session:
x, y = self.makeTestDataFrameWithYAsDataFrame() x, y = self.makeTestDataFrameWithYAsDataFrame()
y = y.rename(columns={'a_target': 'a', 'b_target': 'a_n'}) y = y.rename(columns={'a_target': 'a', 'b_target': 'a_n'})
input_fn = pandas_io.pandas_input_fn( input_fn = pandas_io.pandas_input_fn(
x, y, batch_size=2, shuffle=False, num_epochs=1) x, y, batch_size=2, shuffle=False, num_epochs=1)
features, targets = self.callInputFnOnce(input_fn, session) features, targets = self.callInputFnOnce(input_fn, session)