การกำหนด Feature Columns ใน TensorFlow ด้วย Python

แนวคิด

สร้าง feature columns สำหรับ Estimator โดยวนลูป keys ของ training dataset สร้าง numeric column

ใช้ Keras Sequential API, CNN, TensorFlow Text, Google Colab, Estimator เหมือนเดิม

ตัวอย่างโค้ด

print("Building list of feature columns for estimator model")
my_feature_columns = []
for key in train.keys():
    my_feature_columns.append(tf.feature_column.numeric_column(key=key))

Credit: TensorFlow Tutorials

ผลลัพธ์

Building list of feature columns for estimator model

คำอธิบาย

• Feature column อธิบายว่าโมเดลควรใช้ raw input data จาก features dictionary อย่างไร
• สร้าง Estimator ต้องส่ง list ของ feature columns เข้าไป
• tf.feature_column module มีตัวเลือกมากมายสำหรับ represent data
• สร้าง list บอกให้ Estimator ใช้แต่ละ feature เป็น 32-bit float