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
<Kurs>Einführung in Deep Learning mit Python</Kurs>Übungsanweisungen
- Wandle
df.survivedmit der Funktionto_categorical()in eine kategoriale Variable um. - Spezifiziere ein
Sequential-Modell namensmodel. - Füge eine
Dense-Schicht mit32Knoten hinzu. Verwende'relu'alsactivationund(n_cols,)alsinput_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 dieactivation'softmax'sein. - Kompiliere das Modell mit
'sgd'alsoptimizer,'categorical_crossentropy'als Verlustfunktion undmetrics=['accuracy'], um am Ende jeder Epoche die Accuracy (der Anteil korrekter Vorhersagen) zu sehen. - Trainiere das Modell mit
predictorsundtarget.
Interaktive praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# 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
____