{ "name": "UCI_Model", "version": "0.1", "short_description": "UCI Adult Data analysis using Tensorflow for demonstration of Patra Model Cards.", "full_description": "We have trained a ML model using the tensorflow framework to predict income for the UCI Adult Dataset. We leverage this data to run the Patra model cards to capture metadata about the model as well as fairness and explainability metrics.", "keywords": "uci adult, tensorflow, explainability, fairness, patra", "author": "neelk", "input_type": "Tabular", "category": "classification", "citation": "Becker, B. & Kohavi, R. (1996). Adult [Dataset]. UCI Machine Learning Repository. https://doi.org/10.24432/C5XW20.", "input_data": "https://archive.ics.uci.edu/ml/datasets/adult", "output_data": "https://huggingface.co/nkarthikeyan/neelk-uci_model-0.1", "foundational_model": "None", "ai_model": { "name": "Tensorflow Model", "version": "0.1", "description": "Census classification problem using TensorFlow Neural Network using the UCI Adult Dataset", "owner": "Neelesh Karthikeyan", "location": "https://huggingface.co/nkarthikeyan/neelk-uci_model-0.1/blob/main/neelk-uci_model-0.1.h5", "license": "BSD-3 Clause", "framework": "tensorflow", "model_type": "dnn", "test_accuracy": 0.7663135528564453, "inference_label": "", "model_structure": { "module": "keras", "class_name": "Sequential", "config": { "name": "sequential", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" } }, "layers": [ { "module": "keras.layers", "class_name": "InputLayer", "config": { "batch_shape": [ 100 ], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_layer" } }, { "module": "keras.layers", "class_name": "Dense", "config": { "name": "dense", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" } }, "units": 64, "activation": "relu", "use_bias": true, "kernel_initializer": { "module": "keras.initializers", "class_name": "GlorotUniform", "config": {} }, "bias_initializer": { "module": "keras.initializers", "class_name": "Zeros", "config": {} } }, "build_config": { "input_shape": [ 100 ] } }, { "module": "keras.layers", "class_name": "Dense", "config": { "name": "dense_1", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" } }, "units": 128, "activation": "relu", "use_bias": true, "kernel_initializer": { "module": "keras.initializers", "class_name": "GlorotUniform", "config": {} }, "bias_initializer": { "module": "keras.initializers", "class_name": "Zeros", "config": {} } }, "build_config": { "input_shape": [ 64 ] } }, { "module": "keras.layers", "class_name": "Dense", "config": { "name": "dense_2", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" } }, "units": 64, "activation": "relu", "use_bias": true, "kernel_initializer": { "module": "keras.initializers", "class_name": "GlorotUniform", "config": {} }, "bias_initializer": { "module": "keras.initializers", "class_name": "Zeros", "config": {} } }, "build_config": { "input_shape": [ 128 ] } }, { "module": "keras.layers", "class_name": "Dense", "config": { "name": "dense_3", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" } }, "units": 1, "activation": "sigmoid", "use_bias": true, "kernel_initializer": { "module": "keras.initializers", "class_name": "GlorotUniform", "config": {} }, "bias_initializer": { "module": "keras.initializers", "class_name": "Zeros", "config": {} } }, "build_config": { "input_shape": [ 64 ] } } ], "build_input_shape": [ 100 ] }, "build_config": { "input_shape": [ 100 ] }, "compile_config": { "optimizer": { "module": "keras.optimizers", "class_name": "Adam", "config": { "name": "adam", "learning_rate": 0.0010000000474974513, "use_ema": false, "ema_momentum": 0.99, "beta_1": 0.9, "beta_2": 0.999, "epsilon": 1e-07, "amsgrad": false } }, "loss": "binary_crossentropy", "metrics": [ "accuracy" ], "run_eagerly": false, "steps_per_execution": 1, "jit_compile": false } }, "metrics": { "Test loss": 0.5416663885116577, "Epochs": 100, "Batch Size": 32, "Optimizer": "Adam", "Learning Rate": 0.001, "Input Shape": "(26048, 100)" } }, "bias_analysis": { "demographic_parity_diff": 0.007677798575754372, "equal_odds_difference": 0.011425675372248634 }, "xai_analysis": { "fnlwgt": 0.007139739427301619, "capital_gain": 0.007126364939742618, "hours_per_week": 1.1592639817132382e-05, "age": 2.7945637702944362e-06, "relationship__Wife": 1.6353527704876526e-06, "sex__Male": 8.985069062975953e-07, "workclass__Self_emp_not_inc": 2.0090076658497835e-07, "workclass__State_gov": 7.496939765063962e-08, "native_country__Holand_Netherlands": 0.0, "native_country__Haiti": 0.0 }, "model_requirements": [ "absl-py==2.1.0", "aiohappyeyeballs==2.5.0", "aiohttp==3.11.13", "aiosignal==1.3.2", "annotated-types==0.7.0", "anyio==4.8.0", "appnope==0.1.4", "argon2-cffi-bindings==21.2.0", "argon2-cffi==23.1.0", "arrow==1.3.0", "asttokens==3.0.0", "astunparse==1.6.3", "async-lru==2.0.4", "attrs==23.1.0", "autocommand==2.2.2", "babel==2.17.0", "backports.tarfile==1.2.0", "beautifulsoup4==4.13.3", "bleach==6.2.0", "boto3==1.36.26", "botocore==1.36.26", "certifi==2025.1.31", "cffi==1.17.1", "charset-normalizer==3.4.1", "click==8.1.8", "cloudpickle==3.1.1", "comm==0.2.2", "contourpy==1.3.1", "cryptography==44.0.2", "cycler==0.12.1", "debugpy==1.8.12", "decorator==5.2.0", "defusedxml==0.7.1", "deprecated==1.2.18", "distro==1.9.0", "executing==2.2.0", "fastjsonschema==2.21.1", "filelock==3.17.0", "filetype==1.2.0", "fire==0.7.0", "flatbuffers==25.2.10", "fonttools==4.56.0", "fqdn==1.5.1", "frozenlist==1.5.0", "fsspec==2025.2.0", "gast==0.6.0", "gitdb==4.0.12", "gitpython==3.1.44", "google-pasta==0.2.0", "grpcio==1.70.0", "h11==0.14.0", "h5py==3.13.0", "httpcore==1.0.7", "httpx==0.28.1", "huggingface-hub==0.24.7", "idna==3.7", "importlib-metadata==8.0.0", "inflect==7.3.1", "iniconfig==2.0.0", "ipykernel==6.29.5", "ipython==8.32.0", "isoduration==20.11.0", "jaraco.collections==5.1.0", "jaraco.context==5.3.0", "jaraco.functools==4.0.1", "jaraco.text==3.12.1", "jedi==0.19.2", "jinja2==3.1.5", "jiter==0.8.2", "jmespath==1.0.1", "joblib==1.4.2", "json5==0.10.0", "jsonpointer==3.0.0", "jsonschema-specifications==2024.10.1", "jsonschema==4.18.6", "jupyter-client==8.6.3", "jupyter-core==5.7.2", "jupyter-events==0.12.0", "jupyter-lsp==2.2.5", "jupyter-server-terminals==0.5.3", "jupyter-server==2.15.0", "jupyterlab-pygments==0.3.0", "jupyterlab-server==2.27.3", "jupyterlab==4.3.5", "keras==3.9.0", "kiwisolver==1.4.8", "libclang==18.1.1", "llvmlite==0.44.0", "markdown-it-py==3.0.0", "markdown==3.7", "markupsafe==3.0.2", "matplotlib-inline==0.1.7", "matplotlib==3.10.0", "mdurl==0.1.2", "mistune==3.1.2", "ml-dtypes==0.4.1", "more-itertools==10.3.0", "mpmath==1.3.0", "multidict==6.1.0", "namex==0.0.8", "nbclient==0.10.2", "nbconvert==7.16.6", "nbformat==5.10.4", "nest-asyncio==1.6.0", "networkx==3.4.2", "notebook-shim==0.2.4", "notebook==7.3.2", "numba==0.61.0", "numpy==1.26.4", "openai==0.28.0", "opencv-python-headless==4.10.0.84", "opencv-python==4.10.0.84", "opt-einsum==3.4.0", "optree==0.14.1", "overrides==7.7.0", "packaging==24.2", "pandas==2.2.3", "pandocfilters==1.5.1", "parso==0.8.4", "patra-toolkit==0.1.2", "pexpect==4.9.0", "pillow==11.1.0", "pip==23.2.1", "platformdirs==4.3.6", "pluggy==1.5.0", "prometheus-client==0.21.1", "prompt-toolkit==3.0.50", "propcache==0.3.0", "protobuf==5.29.3", "psutil==7.0.0", "ptyprocess==0.7.0", "pure-eval==0.2.3", "py-cpuinfo==9.0.0", "pybboxes==0.1.6", "pycparser==2.22", "pydantic-core==2.27.2", "pydantic==2.10.6", "pygithub==2.6.1", "pygments==2.19.1", "pyjwt==2.10.1", "pynacl==1.5.0", "pyparsing==3.2.1", "pyrsistent==0.19.3", "pytest==8.3.4", "python-dateutil==2.9.0.post0", "python-dotenv==1.0.1", "python-json-logger==3.2.1", "pytz==2025.1", "pyyaml==6.0.2", "pyzmq==26.2.1", "referencing==0.36.2", "requests-toolbelt==1.0.0", "requests==2.32.3", "rfc3339-validator==0.1.4", "rfc3986-validator==0.1.1", "rich==13.9.4", "roboflow==1.1.54", "rpds-py==0.22.3", "s3transfer==0.11.2", "sahi==0.11.20", "scikit-learn==1.5.2", "scipy==1.13.1", "seaborn==0.13.2", "send2trash==1.8.3", "setuptools==75.4.0", "shapely==2.0.7", "six==1.17.0", "slicer==0.0.8", "smmap==5.0.2", "sniffio==1.3.1", "soupsieve==2.6", "stack-data==0.6.3", "sympy==1.13.1", "tensorboard-data-server==0.7.2", "tensorboard==2.18.0", "tensorflow-io-gcs-filesystem==0.37.1", "tensorflow==2.18.0", "termcolor==2.5.0", "terminado==0.18.1", "terminaltables==3.1.10", "thop==0.1.1.post2209072238", "threadpoolctl==3.5.0", "tinycss2==1.4.0", "tomli==2.0.1", "torch==2.6.0", "torchvision==0.21.0", "tornado==6.4.2", "tqdm==4.67.1", "traitlets==5.14.3", "typeguard==4.3.0", "types-python-dateutil==2.9.0.20241206", "typing-extensions==4.12.2", "tzdata==2025.1", "ultralytics-thop==2.0.14", "ultralytics==8.3.78", "uri-template==1.3.0", "urllib3==2.3.0", "wcwidth==0.2.13", "webcolors==24.11.1", "webencodings==0.5.1", "websocket-client==1.8.0", "werkzeug==3.1.3", "wheel==0.44.0", "wrapt==1.17.2", "yarl==1.18.3", "yolov5==7.0.14", "zipp==3.19.2" ], "id": "neelk-uci_model-0.1" }