Merge pull request pycaret#853 from batmanscode/patch-1

hxkoey · Nov 17, 2020 · 7040f01 · 7040f01
2 parents 4650c08 + 85f3874
commit 7040f01
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pycaret/arules.py b/pycaret/arules.py
@@ -19,9 +19,9 @@ def setup(data, transaction_id, item_id, ignore_items=None, session_id=None):
     Example
     -------
     >>> from pycaret.datasets import get_data
-    >>> france = get_data('france')
+    >>> data = get_data('france')
     >>> from pycaret.arules import *
-    >>> exp = setup(data = data, transaction_id = 'InvoiceNo', item_id = 'ProductName')
+    >>> exp = setup(data = data, transaction_id = 'InvoiceNo', item_id = 'Description')
         
 
     data: pandas.DataFrame
@@ -126,9 +126,9 @@ def create_model(metric="confidence", threshold=0.5, min_support=0.05, round=4):
     Example
     -------
     >>> from pycaret.datasets import get_data
-    >>> france = get_data('france')
+    >>> data = get_data('france')
     >>> from pycaret.arules import *
-    >>> exp_name = setup(data = data, transaction_id = 'InvoiceNo', item_id = 'ProductName')
+    >>> exp_name = setup(data = data, transaction_id = 'InvoiceNo', item_id = 'Description')
     >>> model1 = create_model(metric = 'confidence')
 
 
@@ -226,8 +226,9 @@ def plot_model(
     Example
     -------
     >>> from pycaret.datasets import get_data
-    >>> france = get_data('france')
+    >>> data = get_data('france')
     >>> from pycaret.arules import *
+    >>> exp_name = setup(data = data, transaction_id = 'InvoiceNo', item_id = 'Description')
     >>> rule1 = create_model(metric='confidence', threshold=0.7, min_support=0.05)
     >>> plot_model(rule1, plot='2d')