ifiguero 2024-12-06 22:53:03 -03:00
parent 7e3320cfda
commit e6fdd013b7
1 changed files with 3 additions and 3 deletions

View File

@ -26,7 +26,7 @@ python train.py
## Estructura del pipeline
La función `load_data` lee el archivo `MODY_data.xlsx` que contiene la base de datos pre, procesada. En caso que no exista debe poseer un archivo de base de datos `HC.xlsx` para generarlo.
La función `load_data` lee el archivo `MODY_data.xlsx` que contiene la base de datos pre-procesada. En caso que no exista debe poseer un archivo de base de datos `HC.xlsx` para generarlo.
```python
df_mody1, df_mody2, df_mody3, df_mody5 = load_data()
@ -35,7 +35,7 @@ df_mody1, df_mody2, df_mody3, df_mody5 = load_data()
Esto devuelve 4 dataframes cada uno conteniendo el dataset completo. Luego creamos un Objeto `BinaryTuner` que contiene las abstracciones para generar los 10 dataset y probar los 10 estrategias de aprendizaje automático. Recibe multiples parametros, siendo el primero el nombre de la columna objetivo, numero de semillas o vector con semillas y proporción del conjunto de pruebas.
```
mody2 = BinaryTuner(dms2, 'MODY2_label', seeds=[231964], drop_ratio=0.2)
mody2 = BinaryTuner(df_mody2, 'MODY2_label', seeds=[231964], drop_ratio=0.2)
```
Esto crea un directorio llamado como la columna objetivo que contiene los registros, modelos e imagenes que se generen en adelante. Esto presenta una restricción si se quiere hacer pruebas con otros dataset con una columna objetivo que tnga el mismo nombre. Se debe tener cuidado verificar que el directorio no existe antes de comenzar un entrenamiento desde cero.
@ -55,4 +55,4 @@ Se generan las graficas SHAP para el modelo/dataset/semilla indicadas.
mody2.wrap_and_save()
```
metodo auxiliar. Crea un zip comprimido con el contenido del directorio asociado. El archivo contiene un timestamp para evitar colisiones en subsecuentes ejecuciones.
metodo auxiliar. Crea un zip comprimido con el contenido del directorio asociado. El archivo contiene un timestamp para evitar colisiones en subsecuentes ejecuciones.