ModelCardGenerator TFX işlem hattı bileşeni, model kartları oluşturur.
Ayrıntılı model kartı formatı için Model Kartı API'sine bakın.
TFX hakkında daha genel bilgi için lütfen TFX Kullanım Kılavuzu'na bakınız.
ModelCardGenerator Bileşenini Yapılandırma
ModelCardGenerator, bir model kartının parçalarını otomatik olarak doldurmak için veri kümesi istatistiklerini , model değerlendirmesini ve itilen modeli alır.
Model kartı alanları ayrıca bir JSON dizesiyle de açıkça doldurulabilir (bu, json modülü kullanılarak oluşturulabilir, aşağıdaki örneğe bakın). Bir alan hem TFX hem de JSON tarafından doldurulursa JSON değeri, TFX değerinin üzerine yazılır.
ModelCardGenerator, model kartı belgelerini kendi yapay çıktısının model_card/ dizinine yazar. model_card.html oluşturmak için kullanılan varsayılan bir HTML model kartı şablonunu kullanır. Özel şablonlar da kullanılabilir; her şablon girişine, template_io argümanında bir dosya adı çıkışı eşlik etmelidir.
Örnek
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')
]
)
Daha fazla ayrıntıyı ModelCardGenerator API referansında bulabilirsiniz.
Tam çalışma örneği için uçtan uca demomuza bakın.