Update README.md

README.md

@@ -32,20 +32,15 @@ from sklearn.model_selection import train_test_split
 
 from tabstar.tabstar_model import TabSTARClassifier
 
-# Load sample data
 csv_path = files("tabstar").joinpath("resources", "imdb.csv")
 x = pd.read_csv(csv_path)
 y = x.pop('Genre_is_Drama')
-
-# Split train/test
 x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.1)
-
-# Initialize and train
+# For regression tasks, replace `TabSTARClassifier` with `TabSTARRegressor`.
 tabstar = TabSTARClassifier()
 tabstar.fit(x_train, y_train)
-
-# Predict and evaluate
-y
+y_pred = tabstar.predict(x_test)
+print(classification_report(y_test, y_pred))
 ```
 
 ---