Hi, I am using the model card toolkit for the generation of model card.
I am using the Scikit-Learn Model Card Toolkit Demo . I run this template on Google Colab, it works fine, but when I run it on my local jyoter notebook, it gives the following error.,
ttributeError: ‘list’ object has no attribute 'train’
This is complete code.
mct = ModelCardToolkit()
model_card = mct.scaffold_assets()
model_card.model_details.name = ‘Breast Cancer Wisconsin (Diagnostic) Dataset’
model_card.model_details.overview = (
'This model predicts whether breast cancer is benign or malignant based on ’
‘image measurements.’)
model_card.model_details.owners = [
{‘name’: ‘Model Cards Team’, ‘contact’: ‘model-cards@google.com’}
]
model_card.model_details.references = [
‘UCI Machine Learning Repository: Breast Cancer Wisconsin (Diagnostic) Data Set’,
‘https://minds.wisconsin.edu/bitstream/handle/1793/59692/TR1131.pdf’
]
model_card.model_details.version.name = str(uuid.uuid4())
model_card.model_details.version.date = str(date.today())
model_card.considerations.ethical_considerations = [{
‘name’: ('Manual selection of image sections to digitize could create ’
‘selection bias’),
‘mitigation_strategy’: ‘Automate the selection process’
}]
model_card.considerations.limitations = [‘Breast cancer diagnosis’]
model_card.considerations.use_cases = [‘Breast cancer diagnosis’]
model_card.considerations.users = [‘Medical professionals’, ‘ML researchers’]
print(type(model_card.model_parameters.data))
print(dir(model_card.model_parameters.data))
model_card.model_parameters.data.train.graphics.description = (
f’{len(X_train)} rows with {len(X_train.columns)} features’)
model_card.model_parameters.data.train.graphics.collection = [
{‘image’: mean_radius_train},
{‘image’: mean_texture_train}
]
model_card.model_parameters.data.eval.graphics.description = (
f’{len(X_test)} rows with {len(X_test.columns)} features’)
model_card.model_parameters.data.eval.graphics.collection = [
{‘image’: mean_radius_test},
{‘image’: mean_texture_test}
]
model_card.quantitative_analysis.graphics.description = (
‘ROC curve and confusion matrix’)
model_card.quantitative_analysis.graphics.collection = [
{‘image’: roc_curve},
{‘image’: confusion_matrix}
]
mct.update_model_card_json(model_card)
AttributeError Traceback (most recent call last)
in
25 print(dir(model_card.model_parameters.data))
26 model_card.model_parameters.data.train.graphics.description = (
—> 27 f’{len(X_train)} rows with {len(X_train.columns)} features’)
28 model_card.model_parameters.data.train.graphics.collection = [
29 {‘image’: mean_radius_train},
AttributeError: ‘list’ object has no attribute ‘train’