Fixed the logistic regression code as well.
This commit is contained in:
parent
1eb6609e9f
commit
bc377aa9fa
2 changed files with 64 additions and 4 deletions
|
|
@ -211,8 +211,9 @@ if __name__ == "__main__":
|
|||
|
||||
# check for weak correlation with target --> worsts have the most impact
|
||||
target_corr = df.corr()['Diagnosis'].abs().sort_values(ascending=False)
|
||||
print("Correlation with target variable descending order:")
|
||||
print("\nCorrelation with target variable descending order:")
|
||||
print(target_corr)
|
||||
print("") # \n splitter
|
||||
|
||||
rng = np.random.default_rng(seed=42)
|
||||
n_train = len(df)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue