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:
stingray0225
2026-01-09 03:18:53 +09:00
parent 3e15c17ace
commit 9c412771ad
3 changed files with 11165 additions and 2 deletions
+2 -2
View File
@@ -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))]),