1 回答

TA貢獻(xiàn)1784條經(jīng)驗(yàn) 獲得超8個(gè)贊
我讓它工作了。我的問題是,在數(shù)據(jù)集中,標(biāo)簽(文檔類別)是字符串格式(例如:“sport”、“business”……)。因此,為了將其編碼為整數(shù),我使用了 Transform 組件對(duì)其進(jìn)行預(yù)處理。
但是,在構(gòu)建評(píng)估器組件時(shí),我傳遞了 ExampleGen 組件,其中沒有對(duì)數(shù)據(jù)進(jìn)行任何處理。所以評(píng)估者試圖從 ExampleGen 轉(zhuǎn)換字符串以匹配模型的整數(shù)輸出。
所以,為了解決這個(gè)問題,我只是這樣做了:
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)
我使用了轉(zhuǎn)換組件中的示例。當(dāng)然,我還更改了配置中的標(biāo)簽鍵,以匹配轉(zhuǎn)換組件的標(biāo)簽名稱。
我不知道是否有一種“更清潔”的方法來(lái)執(zhí)行此操作(或者如果我做錯(cuò)了,請(qǐng)糾正我?。?/p>
添加回答
舉報(bào)