Rangkuman: Penyortir Benih

Termasuk bagian Pembelajaran Terawasi dalam perjalanan Machine Learning di Coddy. Pelajaran 5 dari 58.

Tantangan

Mudah

Ingat benih yang membuka bab ini? Sekarang buat pengklasifikasinya. seeds.csv berisi 180 benih gandum: empat pengukuran yang disebutkan dalam features, dan variety. Kode awal membaca benih baru ke dalam tabel satu baris new_seed.

Bagi data, sisihkan seperempat untuk pengujian dengan random_state=0, lalu cetak jumlahnya.
Cetak varietas yang paling umum dalam data pelatihan: jawaban baseline.
Latih model k-nearest neighbors menggunakan pengukuran tersebut. Cetak akurasi model dan baseline, dibulatkan menjadi 3 angka desimal.
Cetak varietas benih baru.

Untuk benih dari perkebunan lain:

Train: 120 seeds, test: 40
Most common in training: Rosa
Accuracy: 0.925 (baseline 0.4)
New seed: Canadian

Coba sendiri

import pandas as pd

# 180 biji gandum: empat ukuran dan varietasnya
seeds = pd.read_csv("seeds.csv")
features = ["area_mm2", "length_mm", "width_mm", "groove_mm"]

# Biji baru dari laboratorium, dalam tabel satu baris
new_seed = pd.DataFrame({
    "area_mm2": [float(input())],
    "length_mm": [float(input())],
    "width_mm": [float(input())],
    "groove_mm": [float(input())],
})

# Tulis kodemu di bawah ini