Model.predict

In [ ]:
Model.predict(vdf: (str, vDataFrame),
              X: list = [],
              name: str = "",
              cutoff: float = -1,
              pos_label = None,
              inplace: bool = True)

Predicts using the input relation.

Parameters

Name Type Optional Description
vdf
str / vDataFrame
❌
Object to use to run the prediction. It can also be a customized relation but you need to englobe it using an alias. For example "(SELECT 1) x" is correct whereas "(SELECT 1)" or "SELECT 1" are incorrect.
X
list
✓
List of the columns used to deploy the models. If empty, the model predictors will be used.
name
str
✓
Name of the added vcolumn. If empty, a name will be generated.
cutoff
float
✓
Cutoff for which the tested category will be accepted as prediction. If the parameter is not between 0 and 1, the class probability will be returned.
pos_label
int / float / str
✓
Class label.
inplace
bool
✓
If set to True, the prediction will be added to the vDataFrame.

Returns

vDataFrame : the input object.

Example

In [28]:
from verticapy import vDataFrame
iris = vDataFrame("public.iris")
display(iris)
123
SepalLengthCm
Numeric(5,2)
123
SepalWidthCm
Numeric(5,2)
123
PetalLengthCm
Numeric(5,2)
123
PetalWidthCm
Numeric(5,2)
Abc
Species
Varchar(30)
14.303.001.100.10Iris-setosa
24.402.901.400.20Iris-setosa
34.403.001.300.20Iris-setosa
44.403.201.300.20Iris-setosa
54.502.301.300.30Iris-setosa
64.603.101.500.20Iris-setosa
74.603.201.400.20Iris-setosa
84.603.401.400.30Iris-setosa
94.603.601.000.20Iris-setosa
104.703.201.300.20Iris-setosa
114.703.201.600.20Iris-setosa
124.803.001.400.10Iris-setosa
134.803.001.400.30Iris-setosa
144.803.101.600.20Iris-setosa
154.803.401.600.20Iris-setosa
164.803.401.900.20Iris-setosa
174.902.403.301.00Iris-versicolor
184.902.504.501.70Iris-virginica
194.903.001.400.20Iris-setosa
204.903.101.500.10Iris-setosa
214.903.101.500.10Iris-setosa
224.903.101.500.10Iris-setosa
235.002.003.501.00Iris-versicolor
245.002.303.301.00Iris-versicolor
255.003.001.600.20Iris-setosa
265.003.201.200.20Iris-setosa
275.003.301.400.20Iris-setosa
285.003.401.500.20Iris-setosa
295.003.401.600.40Iris-setosa
305.003.501.300.30Iris-setosa
315.003.501.600.60Iris-setosa
325.003.601.400.20Iris-setosa
335.102.503.001.10Iris-versicolor
345.103.301.700.50Iris-setosa
355.103.401.500.20Iris-setosa
365.103.501.400.20Iris-setosa
375.103.501.400.30Iris-setosa
385.103.701.500.40Iris-setosa
395.103.801.500.30Iris-setosa
405.103.801.600.20Iris-setosa
415.103.801.900.40Iris-setosa
425.202.703.901.40Iris-versicolor
435.203.401.400.20Iris-setosa
445.203.501.500.20Iris-setosa
455.204.101.500.10Iris-setosa
465.303.701.500.20Iris-setosa
475.403.004.501.50Iris-versicolor
485.403.401.500.40Iris-setosa
495.403.401.700.20Iris-setosa
505.403.701.500.20Iris-setosa
515.403.901.300.40Iris-setosa
525.403.901.700.40Iris-setosa
535.502.304.001.30Iris-versicolor
545.502.403.701.00Iris-versicolor
555.502.403.801.10Iris-versicolor
565.502.504.001.30Iris-versicolor
575.502.604.401.20Iris-versicolor
585.503.501.300.20Iris-setosa
595.504.201.400.20Iris-setosa
605.602.503.901.10Iris-versicolor
615.602.704.201.30Iris-versicolor
625.602.804.902.00Iris-virginica
635.602.903.601.30Iris-versicolor
645.603.004.101.30Iris-versicolor
655.603.004.501.50Iris-versicolor
665.702.505.002.00Iris-virginica
675.702.603.501.00Iris-versicolor
685.702.804.101.30Iris-versicolor
695.702.804.501.30Iris-versicolor
705.702.904.201.30Iris-versicolor
715.703.004.201.20Iris-versicolor
725.703.801.700.30Iris-setosa
735.704.401.500.40Iris-setosa
745.802.604.001.20Iris-versicolor
755.802.703.901.20Iris-versicolor
765.802.704.101.00Iris-versicolor
775.802.705.101.90Iris-virginica
785.802.705.101.90Iris-virginica
795.802.805.102.40Iris-virginica
805.804.001.200.20Iris-setosa
815.903.004.201.50Iris-versicolor
825.903.005.101.80Iris-virginica
835.903.204.801.80Iris-versicolor
846.002.204.001.00Iris-versicolor
856.002.205.001.50Iris-virginica
866.002.705.101.60Iris-versicolor
876.002.904.501.50Iris-versicolor
886.003.004.801.80Iris-virginica
896.003.404.501.60Iris-versicolor
906.102.605.601.40Iris-virginica
916.102.804.001.30Iris-versicolor
926.102.804.701.20Iris-versicolor
936.102.904.701.40Iris-versicolor
946.103.004.601.40Iris-versicolor
956.103.004.901.80Iris-virginica
966.202.204.501.50Iris-versicolor
976.202.804.801.80Iris-virginica
986.202.904.301.30Iris-versicolor
996.203.405.402.30Iris-virginica
1006.302.304.401.30Iris-versicolor
Rows: 1-100 of 150 | Columns: 5
In [29]:
# Multiclass Classification
from verticapy.learn.naive_bayes import MultinomialNB
model = MultinomialNB("MultinomialNB_iris")
model.drop()
model.fit("public.iris", 
          ["SepalLengthCm", "SepalWidthCm"], 
          "Species")
# Probability Iris Setosa
model.predict(iris, 
              X = ["SepalLengthCm", "SepalWidthCm"],
              name = "prediction_proba_setosa",
              pos_label = "Iris-setosa")
123
SepalLengthCm
Numeric(5,2)
123
SepalWidthCm
Numeric(5,2)
123
PetalLengthCm
Numeric(5,2)
123
PetalWidthCm
Numeric(5,2)
Abc
Species
Varchar(30)
Abc
prediction_proba_setosa
Varchar(100)
14.303.001.100.10Iris-setosa0.934314
24.402.901.400.20Iris-setosa0.893412
34.403.001.300.20Iris-setosa0.930579
44.403.201.300.20Iris-setosa0.973616
54.502.301.300.30Iris-setosa0.446651
64.603.101.500.20Iris-setosa0.944621
74.603.201.400.20Iris-setosa0.966315
84.603.401.400.30Iris-setosa0.988582
94.603.601.000.20Iris-setosa0.996463
104.703.201.300.20Iris-setosa0.959346
114.703.201.600.20Iris-setosa0.959346
124.803.001.400.10Iris-setosa0.870361
134.803.001.400.30Iris-setosa0.870361
144.803.101.600.20Iris-setosa0.91683
154.803.401.600.20Iris-setosa0.982427
164.803.401.900.20Iris-setosa0.982427
174.902.403.301.00Iris-versicolor0.312809
184.902.504.501.70Iris-virginica0.387397
194.903.001.400.20Iris-setosa0.834521
204.903.101.500.10Iris-setosa0.892174
214.903.101.500.10Iris-setosa0.892174
224.903.101.500.10Iris-setosa0.892174
235.002.003.501.00Iris-versicolor0.107096
245.002.303.301.00Iris-versicolor0.19585
255.003.001.600.20Iris-setosa0.783541
265.003.201.200.20Iris-setosa0.908687
275.003.301.400.20Iris-setosa0.944638
285.003.401.500.20Iris-setosa0.967588
295.003.401.600.40Iris-setosa0.967588
305.003.501.300.30Iris-setosa0.981551
315.003.501.600.60Iris-setosa0.981551
325.003.601.400.20Iris-setosa0.989738
335.102.503.001.10Iris-versicolor0.239025
345.103.301.700.50Iris-setosa0.920841
355.103.401.500.20Iris-setosa0.953058
365.103.501.400.20Iris-setosa0.973037
375.103.501.400.30Iris-setosa0.973037
385.103.701.500.40Iris-setosa0.991719
395.103.801.500.30Iris-setosa0.995538
405.103.801.600.20Iris-setosa0.995538
415.103.801.900.40Iris-setosa0.995538
425.202.703.901.40Iris-versicolor0.303069
435.203.401.400.20Iris-setosa0.929354
445.203.501.500.20Iris-setosa0.958854
455.204.101.500.10Iris-setosa0.998997
465.303.701.500.20Iris-setosa0.97906
475.403.004.501.50Iris-versicolor0.377512
485.403.401.500.40Iris-setosa0.826533
495.403.401.700.20Iris-setosa0.826533
505.403.701.500.20Iris-setosa0.964329
515.403.901.300.40Iris-setosa0.989338
525.403.901.700.40Iris-setosa0.989338
535.502.304.001.30Iris-versicolor0.0239515
545.502.403.701.00Iris-versicolor0.0318631
555.502.403.801.10Iris-versicolor0.0318631
565.502.504.001.30Iris-versicolor0.0434823
575.502.604.401.20Iris-versicolor0.0606972
585.503.501.300.20Iris-setosa0.822746
595.504.201.400.20Iris-setosa0.997196
605.602.503.901.10Iris-versicolor0.0244344
615.602.704.201.30Iris-versicolor0.0492292
625.602.804.902.00Iris-virginica0.0719254
635.602.903.601.30Iris-versicolor0.106317
645.603.004.101.30Iris-versicolor0.157569
655.603.004.501.50Iris-versicolor0.157569
665.702.505.002.00Iris-virginica0.0130066
675.702.603.501.00Iris-versicolor0.0182952
685.702.804.101.30Iris-versicolor0.0388047
695.702.804.501.30Iris-versicolor0.0388047
705.702.904.201.30Iris-versicolor0.0581129
715.703.004.201.20Iris-versicolor0.0880216
725.703.801.700.30Iris-setosa0.872634
735.704.401.500.40Iris-setosa0.997111
745.802.604.001.20Iris-versicolor0.00923874
755.802.703.901.20Iris-versicolor0.0133147
765.802.704.101.00Iris-versicolor0.0133147
775.802.705.101.90Iris-virginica0.0133147
785.802.705.101.90Iris-virginica0.0133147
795.802.805.102.40Iris-virginica0.0196354
805.804.001.200.20Iris-setosa0.914741
815.903.004.201.50Iris-versicolor0.0215856
825.903.005.101.80Iris-virginica0.0215856
835.903.204.801.80Iris-versicolor0.0529649
846.002.204.001.00Iris-versicolor0.00062211
856.002.205.001.50Iris-virginica0.00062211
866.002.705.101.60Iris-versicolor0.00287994
876.002.904.501.50Iris-versicolor0.00631716
886.003.004.801.80Iris-virginica0.00965039
896.003.404.501.60Iris-versicolor0.0610597
906.102.605.601.40Iris-virginica0.000868411
916.102.804.001.30Iris-versicolor0.00179776
926.102.804.701.20Iris-versicolor0.00179776
936.102.904.701.40Iris-versicolor0.00267357
946.103.004.601.40Iris-versicolor0.00405697
956.103.004.901.80Iris-virginica0.00405697
966.202.204.501.50Iris-versicolor0.000114371
976.202.804.801.80Iris-virginica0.000724077
986.202.904.301.30Iris-versicolor0.00106772
996.203.405.402.30Iris-virginica0.00983732
1006.302.304.401.30Iris-versicolor5.7715e-05
Out[29]:
Rows: 1-100 of 150 | Columns: 6
In [30]:
# Global Prediction
model.predict(iris, 
              X = ["SepalLengthCm", "SepalWidthCm"],
              name = "prediction")
123
SepalLengthCm
Numeric(5,2)
123
SepalWidthCm
Numeric(5,2)
123
PetalLengthCm
Numeric(5,2)
123
PetalWidthCm
Numeric(5,2)
Abc
Species
Varchar(30)
Abc
prediction_proba_setosa
Varchar(100)
Abc
prediction
Varchar(100)
14.303.001.100.10Iris-setosa0.934314Iris-setosa
24.402.901.400.20Iris-setosa0.893412Iris-setosa
34.403.001.300.20Iris-setosa0.930579Iris-setosa
44.403.201.300.20Iris-setosa0.973616Iris-setosa
54.502.301.300.30Iris-setosa0.446651Iris-versicolor
64.603.101.500.20Iris-setosa0.944621Iris-setosa
74.603.201.400.20Iris-setosa0.966315Iris-setosa
84.603.401.400.30Iris-setosa0.988582Iris-setosa
94.603.601.000.20Iris-setosa0.996463Iris-setosa
104.703.201.300.20Iris-setosa0.959346Iris-setosa
114.703.201.600.20Iris-setosa0.959346Iris-setosa
124.803.001.400.10Iris-setosa0.870361Iris-setosa
134.803.001.400.30Iris-setosa0.870361Iris-setosa
144.803.101.600.20Iris-setosa0.91683Iris-setosa
154.803.401.600.20Iris-setosa0.982427Iris-setosa
164.803.401.900.20Iris-setosa0.982427Iris-setosa
174.902.403.301.00Iris-versicolor0.312809Iris-versicolor
184.902.504.501.70Iris-virginica0.387397Iris-versicolor
194.903.001.400.20Iris-setosa0.834521Iris-setosa
204.903.101.500.10Iris-setosa0.892174Iris-setosa
214.903.101.500.10Iris-setosa0.892174Iris-setosa
224.903.101.500.10Iris-setosa0.892174Iris-setosa
235.002.003.501.00Iris-versicolor0.107096Iris-versicolor
245.002.303.301.00Iris-versicolor0.19585Iris-versicolor
255.003.001.600.20Iris-setosa0.783541Iris-setosa
265.003.201.200.20Iris-setosa0.908687Iris-setosa
275.003.301.400.20Iris-setosa0.944638Iris-setosa
285.003.401.500.20Iris-setosa0.967588Iris-setosa
295.003.401.600.40Iris-setosa0.967588Iris-setosa
305.003.501.300.30Iris-setosa0.981551Iris-setosa
315.003.501.600.60Iris-setosa0.981551Iris-setosa
325.003.601.400.20Iris-setosa0.989738Iris-setosa
335.102.503.001.10Iris-versicolor0.239025Iris-versicolor
345.103.301.700.50Iris-setosa0.920841Iris-setosa
355.103.401.500.20Iris-setosa0.953058Iris-setosa
365.103.501.400.20Iris-setosa0.973037Iris-setosa
375.103.501.400.30Iris-setosa0.973037Iris-setosa
385.103.701.500.40Iris-setosa0.991719Iris-setosa
395.103.801.500.30Iris-setosa0.995538Iris-setosa
405.103.801.600.20Iris-setosa0.995538Iris-setosa
415.103.801.900.40Iris-setosa0.995538Iris-setosa
425.202.703.901.40Iris-versicolor0.303069Iris-versicolor
435.203.401.400.20Iris-setosa0.929354Iris-setosa
445.203.501.500.20Iris-setosa0.958854Iris-setosa
455.204.101.500.10Iris-setosa0.998997Iris-setosa
465.303.701.500.20Iris-setosa0.97906Iris-setosa
475.403.004.501.50Iris-versicolor0.377512Iris-versicolor
485.403.401.500.40Iris-setosa0.826533Iris-setosa
495.403.401.700.20Iris-setosa0.826533Iris-setosa
505.403.701.500.20Iris-setosa0.964329Iris-setosa
515.403.901.300.40Iris-setosa0.989338Iris-setosa
525.403.901.700.40Iris-setosa0.989338Iris-setosa
535.502.304.001.30Iris-versicolor0.0239515Iris-versicolor
545.502.403.701.00Iris-versicolor0.0318631Iris-versicolor
555.502.403.801.10Iris-versicolor0.0318631Iris-versicolor
565.502.504.001.30Iris-versicolor0.0434823Iris-versicolor
575.502.604.401.20Iris-versicolor0.0606972Iris-versicolor
585.503.501.300.20Iris-setosa0.822746Iris-setosa
595.504.201.400.20Iris-setosa0.997196Iris-setosa
605.602.503.901.10Iris-versicolor0.0244344Iris-versicolor
615.602.704.201.30Iris-versicolor0.0492292Iris-versicolor
625.602.804.902.00Iris-virginica0.0719254Iris-versicolor
635.602.903.601.30Iris-versicolor0.106317Iris-versicolor
645.603.004.101.30Iris-versicolor0.157569Iris-versicolor
655.603.004.501.50Iris-versicolor0.157569Iris-versicolor
665.702.505.002.00Iris-virginica0.0130066Iris-versicolor
675.702.603.501.00Iris-versicolor0.0182952Iris-versicolor
685.702.804.101.30Iris-versicolor0.0388047Iris-versicolor
695.702.804.501.30Iris-versicolor0.0388047Iris-versicolor
705.702.904.201.30Iris-versicolor0.0581129Iris-versicolor
715.703.004.201.20Iris-versicolor0.0880216Iris-versicolor
725.703.801.700.30Iris-setosa0.872634Iris-setosa
735.704.401.500.40Iris-setosa0.997111Iris-setosa
745.802.604.001.20Iris-versicolor0.00923874Iris-versicolor
755.802.703.901.20Iris-versicolor0.0133147Iris-versicolor
765.802.704.101.00Iris-versicolor0.0133147Iris-versicolor
775.802.705.101.90Iris-virginica0.0133147Iris-versicolor
785.802.705.101.90Iris-virginica0.0133147Iris-versicolor
795.802.805.102.40Iris-virginica0.0196354Iris-versicolor
805.804.001.200.20Iris-setosa0.914741Iris-setosa
815.903.004.201.50Iris-versicolor0.0215856Iris-versicolor
825.903.005.101.80Iris-virginica0.0215856Iris-versicolor
835.903.204.801.80Iris-versicolor0.0529649Iris-versicolor
846.002.204.001.00Iris-versicolor0.00062211Iris-versicolor
856.002.205.001.50Iris-virginica0.00062211Iris-versicolor
866.002.705.101.60Iris-versicolor0.00287994Iris-versicolor
876.002.904.501.50Iris-versicolor0.00631716Iris-versicolor
886.003.004.801.80Iris-virginica0.00965039Iris-versicolor
896.003.404.501.60Iris-versicolor0.0610597Iris-virginica
906.102.605.601.40Iris-virginica0.000868411Iris-versicolor
916.102.804.001.30Iris-versicolor0.00179776Iris-versicolor
926.102.804.701.20Iris-versicolor0.00179776Iris-versicolor
936.102.904.701.40Iris-versicolor0.00267357Iris-versicolor
946.103.004.601.40Iris-versicolor0.00405697Iris-versicolor
956.103.004.901.80Iris-virginica0.00405697Iris-versicolor
966.202.204.501.50Iris-versicolor0.000114371Iris-versicolor
976.202.804.801.80Iris-virginica0.000724077Iris-versicolor
986.202.904.301.30Iris-versicolor0.00106772Iris-versicolor
996.203.405.402.30Iris-virginica0.00983732Iris-virginica
1006.302.304.401.30Iris-versicolor5.7715e-05Iris-versicolor
Out[30]:
Rows: 1-100 of 150 | Columns: 7