Published April 10, 2024
| Version 1.0.0
Model
Open
AI Model Collection
Description
Python AI Model classes used in the context of Knowledge EU Project
The models implemented are:
- Autogluon tabular model
- Classifier and Regressor sklearn models, lightgbm, xgboost, catboost
- Tensorflow keras deep learning models using different architectures, pretrained NN and finetuning strategies.
All the models inherit from a general abstract class that implements the following methods:
- fit
- predict
- evaluate
- _load_model
- _save_model
- to_pmml
- from_pmml
- to_onnx
- from_onnx
- get_metadata
All the models can be converted to PMML or ONNX standards.
Files
AI Model Python Classes.zip
Files
(30.1 kB)
Name | Size | Download all |
---|---|---|
md5:becae7cb9888a99acaf87e272d5dc846
|
30.1 kB | Preview Download |
Additional details
Related works
- Is described by
- Publication: 10.3390/systems12020048 (DOI)
- Is part of
- Book chapter: 10.1007/978-3-031-46452-2_19 (DOI)