Bộ công cụ thẻ mô hình

Thư viện Bộ công cụ thẻ mô hình (MCT) sắp xếp hợp lý và tự động hóa việc tạo Thẻ mô hình , tài liệu học máy cung cấp ngữ cảnh và tính minh bạch trong quá trình phát triển và hiệu suất của mô hình. Việc tích hợp Bộ công cụ thẻ mẫu vào đường dẫn ML của bạn sẽ cho phép bạn chia sẻ siêu dữ liệu và chỉ số của mô hình với các nhà nghiên cứu, nhà phát triển, phóng viên, v.v.

MCT lưu trữ các trường thẻ mô hình bằng lược đồ JSON . MCT có thể tự động điền các trường đó cho người dùng TFX thông qua Siêu dữ liệu ML (MLMD) . Các trường thẻ mô hình cũng có thể được điền theo cách thủ công thông qua API Python . Một số trường hợp sử dụng của thẻ mô hình bao gồm:

  • Tạo điều kiện trao đổi thông tin giữa người xây dựng mô hình và người phát triển sản phẩm.
  • Thông báo cho người dùng các mô hình ML để đưa ra quyết định sáng suốt hơn về cách sử dụng chúng (hoặc cách không sử dụng chúng).
  • Cung cấp thông tin mô hình cần thiết để giám sát công hiệu quả và trách nhiệm giải trình.
import model_card_toolkit as mct

# Initialize the Model Card Toolkit with a path to store generate assets
model_card_output_path = ...
toolkit = mct.ModelCardToolkit(model_card_output_path)

# Initialize the ModelCard, which can be freely populated
model_card = toolkit.scaffold_assets()
model_card.model_details.name = 'My Model'

# Write the model card data to a JSON file
toolkit.update_model_card_json(model_card)

# Return the model card document as an HTML page
html = toolkit.export_format()

Tài nguyên