Parameter zum MLproject hinzufügen
Parameter in MLflow Projects zu definieren macht deinen ML-Code reproduzierbar. Parameter erleichtern außerdem das Ausführen von Trainingsexperimenten mit unterschiedlichen Einstellungen, ohne den Code ändern zu müssen.
In dieser Übung fügst du deinem MLproject-File Parameter für den main-Entry-Point hinzu. Dieser Entry-Point wird verwendet, um das Skript train_model.py auszuführen, das ein Logistic-Regression-Modell auf Versicherungsdaten trainiert.
Das Skript akzeptiert zwei Parameter, n_jobs und fit_intercept, die als Hyperparameter zum Trainieren des Modells dienen. Du beginnst damit, den Parameter n_jobs in der MLproject-Datei hinzuzufügen. Danach fügst du den Parameter fit_intercept hinzu. Abschließend ergänzt du beide Parameter im Befehl, der im main-Entry-Point ausgeführt wird.
Diese Übung ist Teil des Kurses
Einführung in MLflow
Anleitung zur Übung
- Erstelle einen Parameter namens
n_jobsvom Typintmit dem Standardwert1. - Erstelle einen zweiten Parameter namens
fit_interceptvom Typboolmit dem StandardwertTrue. - Übergib beide Parameter an den Befehl, wobei
n_jobszuerst und anschließendfit_interceptübergeben wird.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
"""
name: insurance_model
python_env: python_env.yaml
entry_points:
main:
parameters:
# Create parameter for number of jobs as n_jobs
____:
____: ____
____: ____
# Create parameter for fit_intercept
____:
____: ____
____: ____
# Add parameters to be passed into the command
command: "python3.9 train_model.py {____} {____}"
"""