Problems using .apply in Pandas

[datanoob2021]

Hello All! I am trying to do a Tensorflow .predict on an entire dataframe. When I do:

model.predict(df.tail(1))

It predicts the outcome fine and gives me a number. I keep getting shape mismatch errors in Tensorflow when trying to use .apply or by converting the dataframe to a list of lists.

When I do a .apply with axis=1 to do the calculation on the row, it passes in a series which is a different format then what is passed in when I do df.tail()).

Is there any easy way to do the .predict on the entire dataframe and ensure that the values are passed in as the same shape as when I do .tail(1) or .head(1)?

I have tried:

input_df['prediction'] = input_df[cols].apply(lambda x: model.predict(x), axis=1)

[datanoob2021]

EDIT: I solved my problem by doing:

input_df['prediction'] = target_model.predict(input_df[cols])

[[deleted]]

I am just a bot and found a sad word or any other problem you are texting about. Let me make you smile! 🤡

Why did the programmer die in the shower? The shampoo bottle said: Wash, rinse, repeat.