Thành phần đường ống đẩy TFX

Thành phần Pusher được sử dụng để thúc đẩy một mô hình xác nhận đến một mục tiêu triển khai trong đào tạo mô hình hoặc tái đào tạo. Trước khi triển khai, Pusher dựa vào một hoặc nhiều phước lành từ các thành phần xác thực khác để quyết định có đẩy mô hình hay không.

• Đánh giá ban phước cho các mô hình nếu mô hình được đào tạo mới là "đủ tốt" để được đẩy vào sản xuất.
• (Không bắt buộc nhưng khuyến khích) InfraValidator ban phước cho các mô hình nếu mô hình này là một cách máy móc thể phân phát trong một môi trường sản xuất.

Một Pusher thành phần tiêu thụ một mô hình đào tạo trong SavedModel định dạng, và tạo ra các SavedModel cùng, cùng với versioning siêu dữ liệu.

Sử dụng thành phần bộ đẩy

Thành phần đường ống Pusher thường rất dễ triển khai và yêu cầu ít tùy chỉnh, vì tất cả công việc được thực hiện bởi thành phần Pusher TFX. Mã điển hình trông như thế này:

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)
  )
)

Đẩy một mô hình được tạo ra từ InfraValidator.

(Từ phiên bản 0.30.0)

InfraValidator cũng có thể sản xuất InfraBlessing vật có chứa một mô hình với hâm lại , và Pusher có thể đẩy nó giống như một Model vật.

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(...)
)

Thông tin chi tiết có sẵn trong các tài liệu tham khảo Pusher API .