def _float_feature(values):
"""Returns a float_list from a float / double."""
return train.Feature(float_list=train.FloatList(value=values))
def _float_feature(value):
"""Returns a float_list from a float / double."""
if not isinstance(value, np.ndarray):
value = [value]
return tf_train.Feature(float_list=tf_train.FloatList(value=value))
def _float_feature(value):
return train.Feature(float_list=train.FloatList(value=[value]))