TensorFlow:如何将tf.Tensor字符串转换为python datetime数据类型?

时间:2018-09-08 11:18:50

标签: python tensorflow

在TensorFlow中,尝试拆分从文件读取的日期值,该日期值已通过数据集api转换为Tensor字符串类型。我想将此Tensor字符串数据类型转换为python datetime类型,以便可以找到星期,小时,天等。

1 个答案:

答案 0 :(得分:1)

这是一个非常小的示例。您可以出于自己的目的对其进行重构。

import tensorflow as tf
from datetime import datetime


sess = tf.Session()

#Should be shortened
def convert_to_date(text):
    date = datetime.strptime(text.decode('ascii'), '%b %d %Y %I:%M%p')
    return date.strftime('%b %d %Y %I:%M%p') 

filenames = ["C:/Machine Learning/text.txt"]

dataset = tf.data.Dataset.from_tensor_slices(filenames)

tf.data.TextLineDataset
dataset = dataset.flat_map(
                   lambda filename :
                   tf.data.TextLineDataset( filename ) ).map( lambda text :
                                                                        tf.py_func(convert_to_date,
                                                                                  [text],
                                                                                  [tf.string]))
iterator = dataset.make_one_shot_iterator()
date = iterator.get_next()


print(sess.run([date]))

输出为

  

[(b'Jun 01 2005 2005:01:33PM')]