From 337cd22711bcea9afc80036f9a900b22d9d00987 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Thu, 16 Jan 2025 13:54:21 -0300 Subject: [PATCH] labels --- load_dataset.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/load_dataset.py b/load_dataset.py index ba9521f..8267e8f 100644 --- a/load_dataset.py +++ b/load_dataset.py @@ -25,7 +25,7 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): if len(groups) != 2: raise ValueError("Target variable must have exactly two unique values") - print (f"groups: {groups}") + glabel = {1: 'MODY Pos', 0: 'MODY Neg'} group1, group2 = groups data_group1 = dfi[dfi[target] == group1][label_columns] data_group2 = dfi[dfi[target] == group2][label_columns] @@ -49,8 +49,8 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): t_stat, t_pval = ttest_ind(group1_values, group2_values, equal_var=False) results.append([ var, "Continua", f"Normal (p={p:.4f})", - f"Media: {mean1:.2f} (Grupo 1), {mean2:.2f} (Grupo 2)", - f"Desviación Est.: {std1:.2f} (Grupo 1), {std2:.2f} (Grupo 2)", + f"Media: {mean1:.2f} ({glabel[group1]}), {mean2:.2f} ({glabel[group1]})", + f"Desviación Est.: {std1:.2f} ({glabel[group1]}), {std2:.2f} ({glabel[group1]})", f"Test t: p={t_pval:.3f}" ]) else: @@ -60,8 +60,8 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): mw_stat, mw_pval = mannwhitneyu(group1_values, group2_values) results.append([ var, "Continua", f"No Normal (p={p:.4f})", - f"Mediana: {median1:.2f} (Grupo 1), {median2:.2f} (Grupo 2)", - f"RIC: {iqr1:.2f} (Grupo 1), {iqr2:.2f} (Grupo 2)", + f"Mediana: {median1:.2f} ({glabel[group1]}), {median2:.2f} ({glabel[group1]})", + f"RIC: {iqr1:.2f} ({glabel[group1]}), {iqr2:.2f} ({glabel[group1]})", f"Mann-Whitney: p={mw_pval:.3f}" ])