diff --git a/load_dataset.py b/load_dataset.py index 2e43808..ba9521f 100644 --- a/load_dataset.py +++ b/load_dataset.py @@ -25,6 +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}") group1, group2 = groups data_group1 = dfi[dfi[target] == group1][label_columns] data_group2 = dfi[dfi[target] == group2][label_columns] @@ -47,7 +48,7 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): mean2, std2 = group2_values.mean(), group2_values.std() t_stat, t_pval = ttest_ind(group1_values, group2_values, equal_var=False) results.append([ - var, "Continua", "Normal", + 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"Test t: p={t_pval:.3f}" @@ -58,7 +59,7 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): median2, iqr2 = group2_values.median(), group2_values.quantile(0.75) - group2_values.quantile(0.25) mw_stat, mw_pval = mannwhitneyu(group1_values, group2_values) results.append([ - var, "Continua", "No Normal", + 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"Mann-Whitney: p={mw_pval:.3f}" @@ -73,7 +74,7 @@ def analisis_univariado(dfi, target=None, continuas=[], discretas=[]): if freq_table.shape[1] == 2: # Test exacto de Fisher _, fisher_pval = fisher_exact(freq_table.values) - test_result = f"Fisher Exact: p={fisher_pval:.3f}" + test_result = f"Fisher Exact: p={fisher_pval:.5f}" else: # Test Chi cuadrado chi2_stat, chi2_pval, _, _ = chi2_contingency(freq_table)