Thêm tham số vào MLproject
Khai báo tham số trong MLflow Projects giúp bạn khiến mã ML có thể tái hiện. Tham số cũng giúp đơn giản hóa việc chạy các thí nghiệm huấn luyện với nhiều thiết lập khác nhau mà không cần sửa mã.
Trong bài tập này, bạn sẽ thêm các tham số vào tệp MLproject cho entry point chính. Entry point này dùng để chạy script train_model.py, script này huấn luyện mô hình Logistic Regression trên dữ liệu Insurance.
Script chấp nhận hai tham số, n_jobs và fit_intercept, là các siêu tham số dùng để huấn luyện mô hình. Bạn sẽ bắt đầu bằng cách thêm tham số n_jobs trong tệp MLproject. Sau đó, bạn sẽ thêm tham số fit_intercept. Cuối cùng, bạn sẽ truyền các tham số này vào lệnh được thực thi ở entry point chính.
Bài tập này là một phần của khóa học
Giới thiệu về MLflow
Hướng dẫn bài tập
- Tạo một tham số tên
n_jobskiểuintvới giá trị mặc định là1. - Tạo tham số thứ hai tên
fit_interceptkiểuboolvới giá trị mặc định làTrue. - Truyền cả hai tham số vào lệnh, đảm bảo
n_jobsđứng trước, tiếp theo làfit_intercept.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
"""
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 {____} {____}"
"""