Tensorflow 2: Как преобразовать несколько строк кадра данных pandas в набор данных элементов последовательности? - PullRequest
0 голосов
/ 02 мая 2020

У меня есть pandas фрейм данных df, данные которого связаны с несколькими событиями. Однако несколько строк в df относятся к каждому событию. Я хотел бы преобразовать фрейм данных в tf.data.Dataset, где каждый элемент соответствует отдельному событию, поэтому каждый элемент набора данных должен соответствовать нескольким строкам df. Затем этот набор данных будет использоваться для обучения RNN.

Я попробовал следующее:

elements = []
for event in df['event_ID'].unique():
    df_event = df[df['event_ID'] == event]
    elements.append(dict(df_event))

ds = tf.data.Dataset.from_tensor_slices(elements)

Однако это приводит к ValueError: Attempt to convert a value with an unsupported type (<class 'dict'>) to a Tensor. Любые идеи, чтобы решить это? Я мог бы преобразовать в numpy массивы, но тогда я больше не могу использовать feature_columns, верно?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...