# Construction in-memory du dataset taxi-fare (équivalent ML.NET ModelInput[])
# Schéma identique au notebook ML.NET : vendor_id, rate_code, passenger_count,
# trip_time_in_secs, trip_distance, payment_type, fare_amount
data = {
"vendor_id": ["CMT", "VST", "CMT", "VST", "CMT", "VST", "CMT", "VST"],
"rate_code": [1.0, np.nan, 1.0, 2.0, 1.0, 1.0, 3.0, 1.0], # NaN = valeur manquante
"passenger_count":[1.0, 1.0, 2.0, 1.0, 3.0, 2.0, 1.0, 4.0],
"trip_time_in_secs":[1271, 474, 2100, 890, 1500, np.nan, 3200, 600],
"trip_distance": [3.8, 1.5, 7.2, 2.9, 5.1, 2.0, 11.0, 1.8],
"payment_type": ["CRD", "CSH", "CRD", "CSH", "CRD", "UNP", "CRD", "CSH"],
"fare_amount": [17.5, 8.0, 28.0, 12.5, 21.0, 9.0, 42.0, 7.5],
}
df = pd.DataFrame(data)
print("DataFrame charge :", df.shape, "lignes x colonnes")
df.head()