Bắt đầu ngayBắt đầu miễn phí

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_jobsfit_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

Xem khóa học

Hướng dẫn bài tập

  • Tạo một tham số tên n_jobs kiểu int với giá trị mặc định là 1.
  • Tạo tham số thứ hai tên fit_intercept kiểu bool vớ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 {____} {____}"
"""
Chỉnh sửa và Chạy Mã