module onnxrt.validate.data

Short summary

module mlprodict.onnxrt.validate.data

Datasets to tests models.

source on GitHub

Functions

function	truncated documentation
load_audit	Use to test conversion of sklearn.ensemble.GradientBoostingClassifier into ONNX.

Documentation

Datasets to tests models.

source on GitHub

mlprodict.onnxrt.validate.data.load_audit()

Use to test conversion of sklearn.ensemble.GradientBoostingClassifier into ONNX.

<<<

from mlprodict.onnxrt.validate.data import load_audit
df = load_audit()
print(df.head())

>>>

       Age Employment  Education  ... IGNORE_Accounts RISK_Adjustment  TARGET_Adjusted
    0   38    Private    College  ...    UnitedStates               0                0
    1   35    Private  Associate  ...         Jamaica               0                0
    2   32    Private     HSgrad  ...    UnitedStates               0                0
    3   45    Private   Bachelor  ...    UnitedStates            7298                1
    4   60    Private    College  ...    UnitedStates           15024                1
    
    [5 rows x 12 columns]

source on GitHub