regression — Turi Create API 6.4.1 documentation (original) (raw)

The Turi Create regression toolkit contains models for regression problems. Currently, we support linear regression and boosted trees. In addition to these models, we provide a smart interface that selects the right model based on the data. If you are unsure about which model to use, simply usecreate() function.

Training data must contain a column for the ‘target’ variable and one or more columns representing feature variables.

Set up the data

import turicreate as tc data = tc.SFrame('https://static.turi.com/datasets/regression/houses.csv')

Select the best model based on your data.

model = tc.regression.create(data, target='price', ... features=['bath', 'bedroom', 'size'])

Make predictions and evaluate results.

predictions = model.predict(data) results = model.evaluate(data)

creating a regression model¶