Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,9 +11,9 @@ def findCorrelation(dataset, target):
|
|
11 |
|
12 |
non_numeric_cols = df.select_dtypes('object').columns.tolist()
|
13 |
|
14 |
-
|
15 |
label_encoder = LabelEncoder()
|
16 |
-
df[non_numeric_col] = label_encoder.fit_transform(df[
|
17 |
|
18 |
d = df.corr()[target].to_dict()
|
19 |
d.pop(target)
|
|
|
11 |
|
12 |
non_numeric_cols = df.select_dtypes('object').columns.tolist()
|
13 |
|
14 |
+
if df[target] in non_numeric_cols:
|
15 |
label_encoder = LabelEncoder()
|
16 |
+
df[non_numeric_col] = label_encoder.fit_transform(df[target])
|
17 |
|
18 |
d = df.corr()[target].to_dict()
|
19 |
d.pop(target)
|