Coverage for mlprodict/onnxrt/validate/data/__init__.py: 100%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1"""
2@file
3@brief Datasets to tests models.
4"""
5import os
6from pandas import read_csv
9def load_audit():
10 """
11 Use to test conversion of
12 :epkg:`sklearn:ensemble:GradientBoostingClassifier`
13 into :epkg:`ONNX`.
15 .. runpython::
16 :showcode:
17 :warningout: DeprecationWarning
19 from mlprodict.onnxrt.validate.data import load_audit
20 df = load_audit()
21 print(df.head())
22 """
23 name = os.path.dirname(__file__)
24 name = os.path.join(name, 'audit.csv')
25 df = read_csv(name).drop(['ID', 'index'], axis=1, inplace=False).dropna()
26 return df