5.6 KiB
5.6 KiB
In [3]:
## prepare env, read and prepare data
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns; sns.set()
codepath = '../2_code' ## for import of user defined module
datapath = '../3_data'
from sys import path; path.insert(1, codepath)
from os import chdir; chdir(datapath)
from bfh_cas_pml import prep_data
X_train, X_test, y_train, y_test = prep_data('bank_data_prep.csv', target='y', seed=1234)In [4]:
## Funktionen (Klassen) importieren
from sklearn.neighbors import KNeighborsClassifier
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import RandomForestClassifier
## tbd ergänzen
from sklearn.metrics import accuracy_score
import time ## für ZeitmessungIn [5]:
## Modelle definieren und in Liste hinterlegen
models = [
KNeighborsClassifier(),
DecisionTreeClassifier(min_impurity_decrease=0.002),
RandomForestClassifier(n_estimators=100)
## tbd ergänzen
]In [6]:
## zum Sammeln der Resultate
scores = []
times_fit = []
times_pred = []
model_names = []
#print('Classifier Score Time fit Time pred')
#print('====================================================================')
## Loop
for model in models:
print(model)
## tbd
## start timer1 - fit - stop timer1
## start timer2 - predict - stop timer2
## berechne Score & pick Modellname
## Ergebnisse an vorbereitete Listen anhängen
## Iterationsergebnisse in Konsole ausgeben (optional)
KNeighborsClassifier() DecisionTreeClassifier(min_impurity_decrease=0.002) RandomForestClassifier()
In [7]:
## visualisieren
## tbd ergänzen