1 回答

TA貢獻1784條經驗 獲得超8個贊
我讓它工作了。我的問題是,在數據集中,標簽(文檔類別)是字符串格式(例如:“sport”、“business”……)。因此,為了將其編碼為整數,我使用了 Transform 組件對其進行預處理。
但是,在構建評估器組件時,我傳遞了 ExampleGen 組件,其中沒有對數據進行任何處理。所以評估者試圖從 ExampleGen 轉換字符串以匹配模型的整數輸出。
所以,為了解決這個問題,我只是這樣做了:
model_resolver = ResolverNode(
instance_name='latest_blessed_model_resolver',
resolver_class=latest_blessed_model_resolver.LatestBlessedModelResolver,
model=Channel(type=Model),
model_blessing=Channel(type=ModelBlessing))
context.run(model_resolver)
evaluator = Evaluator(
examples=transform.outputs['transformed_examples'],
model=trainer.outputs['model'],
baseline_model=model_resolver.outputs['model'],
eval_config=eval_config)
context.run(evaluator)
我使用了轉換組件中的示例。當然,我還更改了配置中的標簽鍵,以匹配轉換組件的標簽名稱。
我不知道是否有一種“更清潔”的方法來執行此操作(或者如果我做錯了,請糾正我?。?/p>
添加回答
舉報