import pandas as pd
from sklearn.datasets import fetch_california_housing
from synthefy_nori import NoriRegressor
X, y = fetch_california_housing(return_X_y=True, as_frame=True)
df = X.copy()
df["price"] = y
target_col = "price"
feature_cols = [c for c in df.columns if c != target_col]
train = df.sample(frac=0.8, random_state=42)
test = df.drop(train.index)
model = NoriRegressor()
model.fit(train[feature_cols].values, train[target_col].values)
predictions = model.predict(test[feature_cols].values)
test["predicted_price"] = predictions
print(test[["price", "predicted_price"]].head())