Model Training with Sklearn#
To use the Sklearn framework, we need to install the extra sklearn dependency in your EvaDB virtual environment.
pip install evadb[sklearn]
2. Example Query#
CREATE OR REPLACE FUNCTION PredictHouseRent FROM ( SELECT number_of_rooms, number_of_bathrooms, days_on_market, rental_price FROM HomeRentals ) TYPE Sklearn PREDICT 'rental_price';
In the above query, you are creating a new customized function by training a model from the
HomeRentals table using the
rental_price column will be the target column for predication, while the rest columns from the
SELECT query are the inputs.