Thẻ mẫu trong TFX

Thành phần quy trình ModelCardGenerator TFX tạo ra các thẻ mô hình.

Để biết định dạng thẻ mẫu chi tiết, hãy xem API thẻ mẫu .

Để biết thêm thông tin chung về TFX, vui lòng xem Hướng dẫn sử dụng TFX .

Định cấu hình Thành phần ModelCardGenerator

ModelCardGenerator lấy số liệu thống kê về tập dữ liệu , đánh giá mô hình và mô hình được đẩy để tự động điền các phần của thẻ mô hình.

Các trường thẻ mẫu cũng có thể được điền rõ ràng bằng chuỗi JSON (có thể tạo chuỗi này bằng mô-đun json , xem Ví dụ bên dưới). Nếu một trường được điền bởi cả TFX và JSON, giá trị JSON sẽ ghi đè giá trị TFX.

ModelCardGenerator ghi các tài liệu thẻ mô hình vào thư mục model_card/ của đầu ra tạo tác của nó. Nó sử dụng mẫu thẻ mô hình HTML mặc định, được sử dụng để tạo model_card.html . Các mẫu tùy chỉnh cũng có thể được sử dụng; mỗi đầu vào mẫu phải đi kèm với đầu ra tên tệp trong template_io arg.

Ví dụ

from model_card_toolkit import ModelCardGenerator
import json

...
model_card_fields = {
  'model_details': {
    'name': 'my_model',
    'owners': 'Google',
    'version': 'v0.1'
  },
  'considerations': {
    'limitations': 'This is a demo model.'
  }
}
mc_gen = ModelCardGenerator(
    statistics=statistics_gen.outputs['statistics'],
    evaluation=evaluator.outputs['evaluation'],
    pushed_model=pusher.outputs['pushed_model'],
    json=json.dumps(model_card_fields),
    template_io=[
        ('html/default_template.html.jinja', 'model_card.html'),
        ('md/default_template.md.jinja', 'model_card.md')
    ]
)

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

Xem bản demo hoàn chỉnh của chúng tôi để biết ví dụ hoạt động đầy đủ.