O Pusher TFX Pipeline Component

Mantenha tudo organizado com as coleções Salve e categorize o conteúdo com base nas suas preferências.

O componente empurrador é utilizado para empurrar um modelo validado para um alvo de implantação durante o treino modelo ou re-formação. Antes da implantação, o Pusher depende de uma ou mais bênçãos de outros componentes de validação para decidir se deve fazer o push do modelo ou não.

  • Avaliador abençoa o modelo se o novo modelo treinado é "bom o suficiente" para ser empurrado para a produção.
  • (Opcional, mas recomendado) InfraValidator abençoa o modelo se o modelo é mecanicamente que possa ser veiculado em um ambiente de produção.

Componente A Pusher consome um modelo treinado em SavedModel formato e produz o mesmo SavedModel, juntamente com o controle de versão de metadados.

Usando o componente Pusher

Um componente de pipeline Pusher é normalmente muito fácil de implantar e requer pouca personalização, já que todo o trabalho é feito pelo componente Pusher TFX. O código típico se parece com isso:

pusher = Pusher(
  model=trainer.outputs['model'],
  model_blessing=evaluator.outputs['blessing'],
  infra_blessing=infra_validator.outputs['blessing'],
  push_destination=tfx.proto.PushDestination(
    filesystem=tfx.proto.PushDestination.Filesystem(
        base_directory=serving_model_dir)
  )
)

Empurrando um modelo produzido a partir do InfraValidator.

(Da versão 0.30.0)

InfraValidator também pode produzir InfraBlessing artefato contendo um modelo com aquecimento , e Pusher pode empurrá-lo apenas como um Model artefato.

infra_validator = InfraValidator(
    ...,
    # make_warmup=True will produce a model with warmup requests in its
    # 'blessing' output.
    request_spec=tfx.proto.RequestSpec(..., make_warmup=True)
)

pusher = Pusher(
    # Push model from 'infra_blessing' input.
    infra_blessing=infra_validator.outputs['blessing'],
    push_destination=tfx.proto.PushDestination(...)
)

Mais detalhes estão disponíveis na referência API Pusher .