![](/img/trans.png)
[英]How to load a TensorRT SavedModel to a TensorFlow Estimator?
[英]Error converting tensorflow estimator to SavedModel
我成功地訓練了一個 TensowFlow 提升樹估計器。 現在我想將它保存為 SavedModel。 問題是我收到以下錯誤。 ValueError: All feature_columns must be _FeatureColumn instances. Given: [NumericColumn(key='fcoeffvariation_Result', shape=(1,), default_value=None, dtype=tf.float32, normalizer_fn=None), NumericColumn(key='fcountabove2sigma_Result', shape=(1,), default_value=None, dtype=tf.float32, normalizer_fn=None)]
下面是我的代碼。 任何人都看到問題是什么? 該measurementData
variabele是熊貓數據幀。 特征fcoeffvariation_Result, fcountabove2sigma_Result
是浮點特征,標簽特征是布爾值(0 或 1)。
measurementData = measurementData[['fcoeffvariation_Result', 'fcountabove2sigma_Result', 'label']]
df = measurementData.copy()
df_features = measurementData.copy()
#Delete target feature from dataframe
del df_features['label']
# Spiting the data to train and test
X_feature = df_features.copy()
Y_label = df['label'].copy()
X_feature_train, X_feature_test, Y_feature_train, Y_feature_test = train_test_split(X_feature, Y_label, test_size=0.3)
############################Create input functions
# Create a input function to train the model
input_func_train = tf.estimator.inputs.pandas_input_fn(x=X_feature_train,y=Y_feature_train, batch_size=50,shuffle=True)
# Create a input function to evaluate the model after train
input_func_test = tf.estimator.inputs.pandas_input_fn(x=X_feature_test, y=Y_feature_test, batch_size=50,shuffle=False)
# Create a input function for prediction
input_func_prediction = tf.estimator.inputs.pandas_input_fn(x=X_feature_test,y=Y_feature_test, batch_size=50,shuffle=False)
###########################Feature Columns
my_feature_columns = [tf.feature_column.numeric_column(key=key)
for key in X_feature_train.keys()]
###########################Train model
linear_est = tf.estimator.LinearClassifier(my_feature_columns)
linear_est.train(input_fn=input_func_train, max_steps=100)
###################################Convert to savedmodel
serving_input_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(
tf.feature_column.make_parse_example_spec([my_feature_columns]))
export_path = linear_est.export_saved_model(
"path", serving_input_fn)
您需要將您的功能轉換為 Tensorflow 可以接受的格式。 在擬合模型之前嘗試將您的列轉換為特征列。 更多信息在這里: https : //www.tensorflow.org/tutorials/structured_data/feature_columns
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.