Python:如何在函数中键入提示 tf.keras 对象？

Question

此示例函数返回 keras 张量的字典:

import pandas as pd
import tensorflow as tf

def create_input_tensors(data: pd.DataFrame) -> Dict[str,tf.keras.engine.keras_tensor.KerasTensor]:
   """Turns each dataframe column into a keras tensor and returns them as a dict"""
   tensors = {}
   for name, column in data.items():
      tensors[name] = tf.keras.Input(shape=(1, ), name=name, dtype=float32)
   return tensors

我不知道如何正确键入提示返回值。运行代码片段会产生以下异常:

Exception has occurred: AttributeError
module 'keras.api._v2.keras' has no attribute 'engine'

谷歌搜索此异常没有帮助。正在运行 type(tensors['year'])在函数末尾的调试器中查看返回字典中的元素之一是什么类型( year 是 data 中的列之一)产生 <class 'keras.engine.keras_tensor.KerasTensor'> .

我对这个特定函数有问题，但通常在尝试键入处理任何类型的 keras 对象的提示函数时也有问题。非常感谢适用于这些类似问题的答案。

Answer 1

这对我来说就像一个魅力:

import typing
from keras.engine.keras_tensor import KerasTensor

def f() -> typing.Dict[str, KerasTensor]:
    return {"a": tf.keras.Input(shape=(1, ),)}

f()