Kartu Model di TFX

Komponen pipa TFX ModelCardGenerator menghasilkan kartu model.

Untuk detail format kartu model, lihat Model Card API .

Untuk informasi lebih umum tentang TFX, silakan lihat Panduan Pengguna TFX .

Mengonfigurasi Komponen ModelCardGenerator

ModelCardGenerator mengambil statistik kumpulan data , evaluasi model , dan model yang didorong untuk secara otomatis mengisi bagian kartu model.

Bidang kartu model juga dapat diisi secara eksplisit dengan string JSON (ini dapat dibuat menggunakan modul json , lihat Contoh di bawah). Jika bidang diisi oleh TFX dan JSON, nilai JSON akan menimpa nilai TFX.

ModelCardGenerator menulis dokumen kartu model ke direktori model_card/ keluaran artefaknya. Ini menggunakan templat kartu model HTML default, yang digunakan untuk menghasilkan model_card.html . Templat khusus juga dapat digunakan; setiap masukan templat harus disertai dengan keluaran nama file di argumen template_io .

Contoh

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

Detail selengkapnya tersedia di referensi API ModelCardGenerator .

Lihat demo menyeluruh kami untuk contoh kerja lengkap.