Letzte Schritte bei Klassifikationsmodellen

Du erstellst jetzt ein Klassifikationsmodell mit dem Titanic-Datensatz, der bereits in ein DataFrame namens df geladen wurde. Du nutzt Informationen über die Passagiere, um vorherzusagen, wer überlebt hat.

Die Prädiktorvariablen sind in einem NumPy-Array predictors gespeichert. Das zu prognostizierende Ziel befindet sich in df.survived, muss für Keras jedoch noch aufbereitet werden. Die Anzahl der Prädiktor-Features ist in n_cols gespeichert.

Hier verwendest du den Optimierer 'sgd', was für Stochastic Gradient Descent steht. Mehr dazu lernst du im nächsten Kapitel!

Diese Übung ist Teil des Kurses

Einführung in Deep Learning mit Python

Kurs anzeigen

Anleitung zur Übung

Wandle df.survived mit der Funktion to_categorical() in eine kategoriale Variable um.
Spezifiziere ein Sequential-Modell namens model.
Füge eine Dense-Schicht mit 32 Knoten hinzu. Verwende 'relu' als activation und (n_cols,) als input_shape.
Füge die Dense-Ausgabeschicht hinzu. Da es zwei Ausgänge gibt, sollte sie zwei Units haben, und weil es ein Klassifikationsmodell ist, sollte die activation 'softmax' sein.
Kompiliere das Modell mit 'sgd' als optimizer, 'categorical_crossentropy' als Verlustfunktion und metrics=['accuracy'], um am Ende jeder Epoche die Accuracy (der Anteil korrekter Vorhersagen) zu sehen.
Trainiere das Modell mit predictors und target.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# Import necessary modules
from tensorflow.keras.layers import Dense
from tensorflow.keras.models import Sequential
from tensorflow.keras.utils import to_categorical

# Convert the target to categorical: target
target = ____

# Set up the model
model = ____

# Add the first layer
____

# Add the output layer
____

# Compile the model
____

# Fit the model
____

Code bearbeiten und ausführen