You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Star0 (0)You must be signed in to star a repository
About
My first ML project in which I first tested multiple models [notebooks/*] on the wellknown Titanic data from Kaggle. The main part consists of an automated and freely configurable Pipeline and analysis of the data. Additionally I implemented a load/dump setup [src/predict.py] to potentially store and later use a previously fitted model.
Dieses Programm kann einen Datensatz analysieren und die Ergebnisse inkl. der Qualität des Fits ausgeben und visualisieren.
Installation:
Für dieses Programm werden lediglich Python inklusive Pandas, Seaborn und SKLearn benötigt. Die Ordner "src" und "data" müssen im selben Ordner liegen.
Der zu analysierende Datensatz muss im Ordner "data" liegen und vom Format .csv sein. Der Dateiname des Datensatzes muss in der "main.py" angepasst werden:
daten=pd.read_csv("../data/**yourdatafile**.csv")
Die Spaltennamen der gewünschten Features müssen in der Variable X eingegeben werden. Der Spaltenname des Targets muss in y eingegeben werden.
#Zur weiteren Verarbeitung der Daten müssen alle numerischen (z.B. Alter, Preis, Hausnummer, Distanz in m, ...) Kategorien in numerical_cats eingegeben werden.
Zur Konfiguration der Analyse gibt es folgende Optionen:
train_test_split kann entsprechend der funktionseigenen Parameter konfiguriert werden. Also Standard sind test_size = 0.25 und random_state = 42 festgelegt.
Das Modell zur Analyse der Daten kann in der Variable pipe festgelegt werden:
Um das gespeicherte Modell zu laden und an weiteren Daten zu testen, kann man die Datei "predict.py" nutzen. Im folgenden Befehl wird der Dateiname des gespeicherten Modells festgelegt:
Die zu testenden Daten können in den übrigen Befehlen, analog zu "main.py" konfiguriert werden.
Beispiel anhand der bekannten Titanic-Daten von Kaggle:
Zu sehen sind die Ausgabe der Feature Importance als Balkendiagramm sowie der Test Data Accuracy als Textausgabe.
Contributors:
Dario (und Props an ChatGPT)
About
My first ML project in which I first tested multiple models [notebooks/*] on the wellknown Titanic data from Kaggle. The main part consists of an automated and freely configurable Pipeline and analysis of the data. Additionally I implemented a load/dump setup [src/predict.py] to potentially store and later use a previously fitted model.