جزء خط لوله ModelCardGenerator TFX کارت های مدل را تولید می کند.
برای جزئیات قالب کارت مدل، به API کارت مدل مراجعه کنید.
برای اطلاعات عمومی بیشتر در مورد TFX، لطفاً راهنمای کاربر TFX را ببینید.
پیکربندی مؤلفه ModelCardGenerator
ModelCardGenerator آمار مجموعه داده ها ، ارزیابی مدل و یک مدل تحت فشار را برای پر کردن خودکار بخش هایی از کارت مدل می گیرد.
فیلدهای کارت مدل را نیز می توان به صراحت با یک رشته JSON پر کرد (این را می توان با استفاده از ماژول json
ایجاد کرد، به مثال زیر مراجعه کنید). اگر یک فیلد با TFX و JSON پر شود، مقدار JSON مقدار TFX را بازنویسی می کند.
ModelCardGenerator اسناد کارت مدل را در پوشه model_card/
خروجی مصنوع خود می نویسد. از یک الگوی کارت مدل HTML پیشفرض استفاده میکند که برای تولید model_card.html
استفاده میشود. همچنین می توان از قالب های سفارشی استفاده کرد. هر ورودی الگو باید با خروجی نام فایل در template_io
arg همراه باشد.
مثال
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')
]
)
جزئیات بیشتر در مرجع ModelCardGenerator API موجود است.
برای نمونه کار کامل ، نسخه ی نمایشی پایان به انتها ما را ببینید.