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
|
from category_encoders import TargetEncoder, OneHotEncoder
|
||||||
|
|
||||||
# CONFIG
|
# CONFIG
|
||||||
TARGET_COL = '' # Target feature's name
|
TARGET_COL = 'deposit' # Target feature's name
|
||||||
CSV_FILE_NAME = '' # CSV file's name
|
CSV_FILE_NAME = 'bank.csv' # CSV file's name
|
||||||
DATA_LEN_LIMIT = 10000 # Limits the number of rows in the data
|
DATA_LEN_LIMIT = 10000 # Limits the number of rows in the data
|
||||||
RANDOM_SEED = 42
|
RANDOM_SEED = 42
|
||||||
MODELS = {"LogReg": Pipeline([("scaler", StandardScaler()), ("clf", LogisticRegression(max_iter=2000))]),
|
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