Các lượt nộp
    Danh sách bài
    Trang chủ
    Báo lỗi

    solution

    Đề bài: [Data Science] Tách tập train/test theo chỉ số

    Để đánh giá mô hình một cách khách quan, ta tách dữ liệu thành tập huấn luyện và tập kiểm tra. Cho nnn giá trị nhãn y0,y1,…,yn−1y_0, y_1, \dots, y_{n-1}y0​,y1​,…,yn−1​ (đánh chỉ số từ 0) và danh sách mmm chỉ số thuộc tập kiểm tra (đôi một khác nhau). Các chỉ số còn lại thuộc tập huấn luyện.

    Hãy in ra số phần tử của mỗi tập và giá trị trung bình của yyy trên mỗi tập. Đảm bảo 1≤m≤n−11 \le m \le n-11≤m≤n−1 nên cả hai tập đều khác rỗng.

    • Định dạng đầu vào:

      Dòng đầu: hai số nguyên nnn mmm. Dòng thứ hai: nnn số thực yiy_iyi​. Dòng thứ ba: mmm số nguyên là các chỉ số (0-based) của tập kiểm tra.

    • Ràng buộc đầu vào:

      2 ≤ n ≤ 10000; 1 ≤ m ≤ n−1; |y| ≤ 10000.

    • Định dạng đầu ra:

      Dòng 1: hai số nguyên — số phần tử tập huấn luyện và số phần tử tập kiểm tra. Dòng 2: hai số thực — trung bình yyy của tập huấn luyện và của tập kiểm tra, làm tròn 2 chữ số thập phân ({:.2f}). Lưu ý: nếu kết quả bằng 0 hãy in 0.00 (tránh in -0.00, có thể cộng thêm 0.0 trước khi in).

    Ví dụ:

    Đầu vào:

    5 2
    10 20 30 40 50
    1 3

    Đầu ra:

    3 2
    30.00 30.00

    Giải thích:

    Tập kiểm tra gồm chỉ số 1, 3 ⇒ y_test = (20, 40), trung bình 30. Tập huấn luyện gồm chỉ số 0, 2, 4 ⇒ y_train = (10, 30, 50), trung bình 30. In: 3 2 rồi 30.00 30.00

    Đang tải editor...