How Good Will Tasters Say This Wine Is?

Walkthrough

1. Load it from the catalogue

   Find the table, read its recorded target and print the column names.

   import pandas as pd
   from welt import Client
   
   with Client() as client:
       item = client.research_datasets(q="white wine quality", limit=1)["items"][0]
       info = client.research_dataset(item["id"])
       data = client.load_research_dataset(item["id"])
   
   target = info["targets"][0]
   target = target["name"] if isinstance(target, dict) else target
   print([column["name"] for column in info["schema"]], target)

2. Pick the rows

   Set aside 20 rows to predict at the end, and keep the rest for the models to learn from.

   known = data.dropna(subset=[target])
   new_rows = known.sample(20, random_state=1)
   table = known.drop(new_rows.index)

3. Compare the models and predict

   The models compete on the same folds against a simple guess, then the top-ranked one is fitted on the whole table. RMSE and MAE are average errors in the target's own units, so lower is better. The last table puts each prediction next to the real answer.

   with Client() as client:
       evaluation = client.evaluate(table, target, task="regression", seed=7,
                                    models=["tabicl-v2", "kumo-medium"])
       model = evaluation.fit()
   
   print(evaluation.summary()[["name", "rank", "rmse_mean", "mae_mean"]])
   features = new_rows.drop(columns=target)
   print(pd.DataFrame({"actual": new_rows[target].values, "predicted": model.predict(features)}))

What you get