| """Exporta el modelo de scikit-learn en formato ONNX: model.onnx. | |
| No vuelve a entrenar: parte de ../sklearn-entrenamiento-evaluacion/model.joblib | |
| (correr antes ese train.py). | |
| Uso: | |
| python exportar.py | |
| """ | |
| from pathlib import Path | |
| import joblib | |
| import numpy as np | |
| from skl2onnx import to_onnx | |
| CARPETA = Path(__file__).parent | |
| modelo = joblib.load(CARPETA.parent / "sklearn-entrenamiento-evaluacion" / "model.joblib") | |
| # ONNX traduce el modelo a un grafo de operaciones. Necesita un ejemplo de entrada | |
| # para saber la forma y el tipo de los datos (3 columnas, float32). | |
| ejemplo = np.zeros((1, 3), dtype="float32") | |
| onx = to_onnx(modelo, ejemplo) | |
| (CARPETA / "model.onnx").write_bytes(onx.SerializeToString()) | |
| print("Modelo guardado en model.onnx") | |