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.