Add bank dataset and results image
Added datasets/bank.csv containing banking data and results/bank.png as an output image. Updated main.py to support the new dataset and generate results.
This commit is contained in:
+11163
File diff suppressed because it is too large
Load Diff
@@ -29,8 +29,8 @@ from sklearn.preprocessing import LabelEncoder
|
||||
from category_encoders import TargetEncoder, OneHotEncoder
|
||||
|
||||
# CONFIG
|
||||
TARGET_COL = '' # Target feature's name
|
||||
CSV_FILE_NAME = '' # CSV file's name
|
||||
TARGET_COL = 'deposit' # Target feature's name
|
||||
CSV_FILE_NAME = 'bank.csv' # CSV file's name
|
||||
DATA_LEN_LIMIT = 10000 # Limits the number of rows in the data
|
||||
RANDOM_SEED = 42
|
||||
MODELS = {"LogReg": Pipeline([("scaler", StandardScaler()), ("clf", LogisticRegression(max_iter=2000))]),
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 209 KiB |
Reference in New Issue
Block a user