Komponent potoku ModelCardGenerator TFX generuje karty modeli.
Aby zapoznać się ze szczegółowym formatem karty modelu, zobacz API karty modelu .
Więcej ogólnych informacji na temat TFX można znaleźć w Podręczniku użytkownika TFX .
Konfigurowanie komponentu ModelCardGenerator
ModelCardGenerator pobiera statystyki zestawu danych , ocenę modelu i wypchnięty model , aby automatycznie wypełnić części karty modelu.
Pola karty modelu można również jawnie wypełnić ciągiem JSON (można to wygenerować za pomocą modułu json , patrz przykład poniżej). Jeśli pole zostanie wypełnione zarówno przez TFX, jak i JSON, wartość JSON zastąpi wartość TFX.
ModelCardGenerator zapisuje dokumenty kart modeli w katalogu model_card/ w wynikach swoich artefaktów. Wykorzystuje domyślny szablon karty modelu HTML, który służy do generowania model_card.html . Można również używać szablonów niestandardowych; każdemu wejściu szablonu musi towarzyszyć wyjściowa nazwa pliku w arg template_io .
Przykład
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')
]
)
Więcej szczegółów można znaleźć w dokumentacji API ModelCardGenerator .
Zobacz nasze kompleksowe demo, aby zapoznać się z pełnym działającym przykładem.