Loading...

verticapy.machine_learning.vertica.tree.DecisionTreeRegressor.predict#

DecisionTreeRegressor.predict(vdf: str | vDataFrame, X: str | list[str] | None = None, name: str | None = None, inplace: bool = True) vDataFrame#

Predicts using the input relation.

Parameters#

vdf: SQLRelation

Object used to run the prediction. You can also specify a customized relation, but you must enclose it with an alias. For example, (SELECT 1) x is valid, whereas (SELECT 1) and SELECT 1 are invalid.

X: SQLColumns, optional

list of the columns used to deploy the models. If empty, the model predictors are used.

name: str, optional

Name of the added :py:class`vDataColumn`. If empty, a name is generated.

inplace: bool, optional

If set to True, the prediction is added to the :py:class`vDataFrame`.

Returns#

vDataFrame

the input object.

Examples#

We import verticapy:

import verticapy as vp

For this example, we will use the winequality dataset.

import verticapy.datasets as vpd

data = vpd.load_winequality()
123
fixed_acidity
Numeric(8)
123
volatile_acidity
Numeric(9)
123
citric_acid
Numeric(8)
123
residual_sugar
Numeric(9)
123
chlorides
Float(22)
123
free_sulfur_dioxide
Numeric(9)
123
total_sulfur_dioxide
Numeric(9)
123
density
Float(22)
123
pH
Numeric(8)
123
sulphates
Numeric(8)
123
alcohol
Float(22)
123
quality
Integer
123
good
Integer
Abc
color
Varchar(20)
13.80.310.0211.10.03620.0114.00.992483.750.4412.460white
23.90.2250.44.20.0329.0118.00.9893.570.3612.881white
34.20.170.361.80.02993.0161.00.989993.650.8912.071white
44.20.2150.235.10.04164.0157.00.996883.420.448.030white
54.40.320.394.30.0331.0127.00.989043.460.3612.881white
64.40.460.12.80.02431.0111.00.988163.480.3413.160white
74.40.540.095.10.03852.097.00.990223.410.412.271white
84.50.190.210.950.03389.0159.00.993323.340.428.050white
94.60.4450.01.40.05311.0178.00.994263.790.5510.250white
104.60.520.152.10.0548.065.00.99343.90.5613.140red
114.70.1450.291.00.04235.090.00.99083.760.4911.360white
124.70.3350.141.30.03669.0168.00.992123.470.4610.550white
134.70.4550.181.90.03633.0106.00.987463.210.8314.071white
144.70.60.172.30.05817.0106.00.99323.850.612.960red
154.70.670.091.00.025.09.00.987223.30.3413.650white
164.70.7850.03.40.03623.0134.00.989813.530.9213.860white
174.80.130.321.20.04240.098.00.98983.420.6411.871white
184.80.170.282.90.0322.0111.00.99023.380.3411.371white
194.80.210.2110.20.03717.0112.00.993243.660.4812.271white
204.80.2250.381.20.07447.0130.00.991323.310.410.360white
214.80.260.2310.60.03423.0111.00.992743.460.2811.571white
224.80.290.231.10.04438.0180.00.989243.280.3411.960white
234.80.330.06.50.02834.0163.00.99373.350.619.950white
244.80.340.06.50.02833.0163.00.99393.360.619.960white
254.80.650.121.10.0134.010.00.992463.320.3613.540white
264.90.2350.2711.750.0334.0118.00.99543.070.59.460white
274.90.330.311.20.01639.0150.00.987133.330.5914.081white
284.90.3350.141.30.03669.0168.00.992123.470.4610.466666666666750white
294.90.3350.141.30.03669.0168.00.992123.470.4610.466666666666750white
304.90.3450.341.00.06832.0143.00.991383.240.410.150white
314.90.3450.341.00.06832.0143.00.991383.240.410.150white
324.90.420.02.10.04816.042.00.991543.710.7414.071red
334.90.470.171.90.03560.0148.00.989643.270.3511.560white
345.00.170.561.50.02624.0115.00.99063.480.3910.871white
355.00.20.41.90.01520.098.00.98973.370.5512.0560white
365.00.2350.2711.750.0334.0118.00.99543.070.59.460white
375.00.240.195.00.04317.0101.00.994383.670.5710.050white
385.00.240.212.20.03931.0100.00.990983.690.6211.760white
395.00.240.341.10.03449.0158.00.987743.320.3213.171white
405.00.2550.222.70.04346.0153.00.992383.750.7611.360white
415.00.270.324.50.03258.0178.00.989563.450.3112.671white
425.00.270.324.50.03258.0178.00.989563.450.3112.671white
435.00.270.41.20.07642.0124.00.992043.320.4710.160white
445.00.290.545.70.03554.0155.00.989763.270.3412.981white
455.00.30.333.70.0354.0173.00.98873.360.313.071white
465.00.310.06.40.04643.0166.00.9943.30.639.960white
475.00.330.161.50.04910.097.00.99173.480.4410.760white
485.00.330.161.50.04910.097.00.99173.480.4410.760white
495.00.330.161.50.04910.097.00.99173.480.4410.760white
505.00.330.184.60.03240.0124.00.991143.180.411.060white
515.00.330.2311.80.0323.0158.00.993223.410.6411.860white
525.00.350.257.80.03124.0116.00.992413.390.411.360white
535.00.350.257.80.03124.0116.00.992413.390.411.360white
545.00.380.011.60.04826.060.00.990843.70.7514.060red
555.00.40.54.30.04629.080.00.99023.490.6613.660red
565.00.420.242.00.0619.050.00.99173.720.7414.081red
575.00.440.0418.60.03938.0128.00.99853.370.5710.260white
585.00.4550.181.90.03633.0106.00.987463.210.8314.071white
595.00.550.148.30.03235.0164.00.99183.530.5112.581white
605.00.610.121.30.00965.0100.00.98743.260.3713.550white
615.00.740.01.20.04116.046.00.992584.010.5912.560red
625.01.020.041.40.04541.085.00.99383.750.4810.540red
635.01.040.241.60.0532.096.00.99343.740.6211.550red
645.10.110.321.60.02812.090.00.990083.570.5212.260white
655.10.140.250.70.03915.089.00.99193.220.439.260white
665.10.1650.225.70.04742.0146.00.99343.180.559.960white
675.10.210.281.40.04748.0148.00.991683.50.4910.450white
685.10.230.181.00.05313.099.00.989563.220.3911.550white
695.10.250.361.30.03540.078.00.98913.230.6412.171white
705.10.260.331.10.02746.0113.00.989463.350.4311.471white
715.10.260.346.40.03426.099.00.994493.230.419.260white
725.10.290.288.30.02627.0107.00.993083.360.3711.060white
735.10.290.288.30.02627.0107.00.993083.360.3711.060white
745.10.30.32.30.04840.0150.00.989443.290.4612.260white
755.10.3050.131.750.03617.073.00.993.40.5112.333333333333350white
765.10.310.30.90.03728.0152.00.9923.540.5610.160white
775.10.330.221.60.02718.089.00.98933.510.3812.571white
785.10.330.221.60.02718.089.00.98933.510.3812.571white
795.10.330.221.60.02718.089.00.98933.510.3812.571white
805.10.330.276.70.02244.0129.00.992213.360.3911.071white
815.10.350.266.80.03436.0120.00.991883.380.411.560white
825.10.350.266.80.03436.0120.00.991883.380.411.560white
835.10.350.266.80.03436.0120.00.991883.380.411.560white
845.10.390.211.70.02715.072.00.98943.50.4512.560white
855.10.420.01.80.04418.088.00.991573.680.7313.671red
865.10.420.011.50.01725.0102.00.98943.380.3612.371white
875.10.470.021.30.03418.044.00.99213.90.6212.860red
885.10.510.182.10.04216.0101.00.99243.460.8712.971red
895.10.520.062.70.05230.079.00.99323.320.439.350white
905.10.5850.01.70.04414.086.00.992643.560.9412.971red
915.20.1550.331.60.02813.059.00.989753.30.8411.981white
925.20.1550.331.60.02813.059.00.989753.30.8411.981white
935.20.160.340.80.02926.077.00.991553.250.5110.160white
945.20.170.270.70.0311.068.00.992183.30.419.850white
955.20.1850.221.00.0347.0123.00.992183.550.4410.1560white
965.20.20.273.20.04716.093.00.992353.440.5310.171white
975.20.210.311.70.04817.061.00.989533.240.3712.071white
985.20.220.466.20.06641.0187.00.993623.190.429.7333333333333350white
995.20.240.157.10.04332.0134.00.993783.240.489.960white
1005.20.240.453.80.02721.0128.00.9923.550.4911.281white
Rows: 1-100 | Columns: 14

Divide your dataset into training and testing subsets.

data = vpd.load_winequality()
train, test = data.train_test_split(test_size = 0.2)

Let’s import the model:

from verticapy.machine_learning.vertica import LinearRegression

Then we can create the model:

model = LinearRegression(
    tol = 1e-6,
    max_iter = 100,
    solver = 'newton',
    fit_intercept = True,
)

We can now fit the model:

model.fit(
    train,
    [
        "fixed_acidity",
        "volatile_acidity",
        "citric_acid",
        "residual_sugar",
        "chlorides",
        "density",
    ],
    "quality",
    test,
)

Prediction is straight-forward:

model.predict(
    test,
    [
        "fixed_acidity",
        "volatile_acidity",
        "citric_acid",
        "residual_sugar",
        "chlorides",
        "density",
    ],
    "prediction",
)
123
fixed_acidity
Numeric(8)
123
volatile_acidity
Numeric(9)
123
citric_acid
Numeric(8)
123
residual_sugar
Numeric(9)
123
chlorides
Float(22)
123
free_sulfur_dioxide
Numeric(9)
123
total_sulfur_dioxide
Numeric(9)
123
density
Float(22)
123
pH
Numeric(8)
123
sulphates
Numeric(8)
123
alcohol
Float(22)
123
quality
Integer
123
good
Integer
Abc
color
Varchar(20)
123
prediction
Float(22)
14.70.4550.181.90.03633.0106.00.987463.210.8314.071white6.2917420535301
24.80.170.282.90.0322.0111.00.99023.380.3411.371white6.15313681866203
35.00.240.212.20.03931.0100.00.990983.690.6211.760white5.99440942293262
45.00.290.545.70.03554.0155.00.989763.270.3412.981white6.25027126730654
55.00.310.06.40.04643.0166.00.9943.30.639.960white5.71206509967371
65.00.330.161.50.04910.097.00.99173.480.4410.760white5.79996815256683
75.00.4550.181.90.03633.0106.00.987463.210.8314.071white6.33322327779837
85.01.040.241.60.0532.096.00.99343.740.6211.550red5.03446258304925
95.10.110.321.60.02812.090.00.990083.570.5212.260white6.19555609382209
105.10.140.250.70.03915.089.00.99193.220.439.260white5.88034452860549
115.10.210.281.40.04748.0148.00.991683.50.4910.450white5.88763001286767
125.10.30.32.30.04840.0150.00.989443.290.4612.260white6.18024043600576
135.10.310.30.90.03728.0152.00.9923.540.5610.160white5.74565383790809
145.10.330.221.60.02718.089.00.98933.510.3812.571white6.15743369721423
155.10.330.221.60.02718.089.00.98933.510.3812.571white6.15743369721423
165.10.520.062.70.05230.079.00.99323.320.439.350white5.52180004367108
175.10.5850.01.70.04414.086.00.992643.560.9412.971red5.51821242283967
185.20.160.340.80.02926.077.00.991553.250.5110.160white5.92536748692689
195.20.170.270.70.0311.068.00.992183.30.419.850white5.83046407456186
205.20.240.157.10.04332.0134.00.993783.240.489.960white5.83725751184585
215.20.310.365.10.03146.0145.00.98973.140.3112.471white6.264719388765
225.20.340.376.20.03142.0133.00.990763.250.4112.560white6.13707266795222
235.20.440.041.40.03643.0119.00.98943.360.3312.181white6.08613927666275
245.20.440.041.40.03643.0119.00.98943.360.3312.181white6.08613927666275
255.20.480.041.60.05419.0106.00.99273.540.6212.271red5.59111632730469
265.20.50.182.00.03623.0129.00.989493.360.7713.471white6.04099769757724
275.20.60.077.00.04433.0147.00.99443.330.589.750white5.48967923028681
285.20.6450.02.150.0815.028.00.994443.780.6112.560red5.24771790248084
295.30.310.3810.50.03153.0140.00.993213.340.4611.760white6.00535610615239
305.30.320.126.60.04322.0141.00.99373.360.610.460white5.78577932778654
315.30.360.276.30.02840.0132.00.991863.370.411.660white5.99311177952728
325.30.3950.071.30.03526.0102.00.9923.50.3510.660white5.75222893839242
335.40.170.272.70.04928.0104.00.992243.460.5510.360white5.93483857236694
345.40.1850.197.10.04836.0110.00.994383.260.419.560white5.81456081716655
355.40.230.361.50.0374.0121.00.989763.240.9912.171white6.18700042063645
365.40.270.224.60.02229.0107.00.988893.330.5413.860white6.43099428437131
375.40.330.314.00.0327.0108.00.990313.30.4312.271white6.14800258157936
385.40.460.152.10.02629.0130.00.989533.390.7713.481white6.09989538866631
395.40.530.162.70.03634.0128.00.988563.20.5313.281white6.21248615242894
405.40.5950.12.80.04226.080.00.99323.360.389.350white5.50926685164853
415.40.740.01.20.04116.046.00.992584.010.5912.560red5.4339067909747
425.50.140.274.60.02922.0104.00.99493.340.449.050white5.67147894840676
435.50.310.293.00.02716.0102.00.990673.230.5611.260white6.08382714807692
445.50.320.454.90.02825.0191.00.99223.510.4911.571white5.92235931939021
455.50.4850.01.50.0658.0103.00.9943.630.49.740white5.44164933065417
465.50.620.331.70.03724.0118.00.987583.150.3913.5560white6.24083027698333
475.60.120.264.30.03818.097.00.994773.360.469.250white5.70624199073842
485.60.130.274.80.02822.0104.00.99483.340.459.260white5.71561195315113
495.60.180.271.70.0331.0103.00.988923.350.3712.960white6.38951171365019
505.60.180.581.250.03429.0129.00.989843.510.612.071white6.20613557697959
515.60.190.270.90.0452.0103.00.990263.50.3911.250white6.15496215793345
525.60.190.474.50.0319.0112.00.99223.560.4511.260white6.01157736984101
535.60.220.321.20.02429.097.00.988233.20.4613.0571white6.43299720462306
545.60.260.2710.60.0327.0119.00.99473.40.3410.771white5.88503095299774
555.60.2950.261.10.03540.0102.00.991543.470.5610.660white5.90455568765475
565.60.320.337.40.03725.095.00.992683.250.4911.160white5.98690081129487
575.60.330.281.20.03133.097.00.991263.490.5810.960white5.92167193619673
585.60.390.244.70.03427.077.00.99063.280.3612.750white6.12744695615461
595.60.410.227.10.0544.0154.00.99313.30.410.550white5.85876903403181
605.60.410.241.90.03410.053.00.988153.320.513.571white6.34388996446791
615.60.420.342.40.02234.097.00.989153.220.3812.871white6.20489732960448
625.60.490.134.50.03917.0116.00.99073.420.913.771white6.04262348336164
635.60.50.092.30.04917.099.00.99373.630.6313.050red5.51356979474687
645.60.50.092.30.04917.099.00.99373.630.6313.050red5.51356979474687
655.60.540.041.70.0495.013.00.99423.720.5811.450red5.39190925106823
665.70.10.271.30.04721.0100.00.99283.270.469.550white5.88672838633047
675.70.150.283.70.04557.0151.00.99133.220.2711.260white6.16806949728917
685.70.150.4711.40.03549.0128.00.994563.030.3410.581white6.01290289537747
695.70.220.216.00.04441.0113.00.998623.220.468.960white5.60456934857149
705.70.220.216.00.04441.0113.00.998623.220.468.960white5.60456934857149
715.70.220.216.00.04441.0113.00.998623.220.468.960white5.60456934857149
725.70.220.281.30.02726.0101.00.989483.350.3812.571white6.27567190721371
735.70.220.293.50.0427.0146.00.989993.170.3612.160white6.29566017207236
745.70.220.331.90.03637.0110.00.989453.260.5812.460white6.3003098108658
755.70.230.257.950.04216.0108.00.994863.440.6110.360white5.78508767529436
765.70.240.476.30.06935.0182.00.993913.110.469.7333333333333350white5.81932804776338
775.70.240.476.30.06935.0182.00.993913.110.469.7550white5.81932804776338
785.70.250.2612.50.04952.5106.00.996913.080.459.460white5.67097280377675
795.70.2550.651.20.07917.0137.00.993073.20.429.450white5.6903298425548
805.70.260.274.10.20173.5189.50.99423.270.389.460white5.68293302071353
815.70.260.274.10.20173.5189.50.99423.270.389.460white5.68293302071353
825.70.2650.286.90.03646.0150.00.992993.360.4410.871white5.9799441476944
835.70.280.2417.50.04460.0167.00.99893.310.449.450white5.5811779920065
845.70.310.284.10.0322.086.00.990623.310.3811.771white6.16694724214668
855.70.320.181.40.02926.0104.00.99063.440.3711.060white6.05647406059308
865.70.330.321.40.04328.093.00.98973.310.512.360white6.16340761594628
875.70.3350.341.00.0413.0174.00.9923.270.6610.050white5.81047081422071
885.70.430.35.70.03924.098.00.9923.540.6112.371white5.9478815042028
895.70.460.461.40.0431.0169.00.99323.130.478.850white5.55204860467126
905.70.6950.066.80.0429.084.00.994323.440.4410.250white5.49354240333594
915.80.130.2212.70.05824.0183.00.99563.320.4211.760white5.97262568905995
925.80.130.265.10.03919.0103.00.994783.360.479.360white5.75959815218081
935.80.150.491.10.04821.098.00.99293.190.489.250white5.81855814875587
945.80.180.371.10.03631.096.00.989423.160.4812.060white6.30903287653638
955.80.230.314.50.04642.0124.00.993243.310.6410.860white5.87646729531775
965.80.240.391.50.05437.0158.00.99323.210.529.360white5.73710114541367
975.80.250.2613.10.05144.0148.00.99723.290.389.350white5.66893792276889
985.80.260.249.20.04455.0152.00.99613.310.389.450white5.65385369802405
995.80.260.249.20.04455.0152.00.99613.310.389.450white5.65385369802405
1005.80.260.291.00.04235.0101.00.990443.360.4811.471white6.10792367074666
Rows: 1-100 | Columns: 15

Important

For this example, a specific model is utilized, and it may not correspond exactly to the model you are working with. To see a comprehensive example specific to your class of interest, please refer to that particular class.