Skip to main content

Python package to mine association rules in datasets

Project description

ruleminer

https://img.shields.io/pypi/v/ruleminer.svg License: MIT Code style: black

DISCLAIMER - BETA PHASE

This package is currently in a beta phase.

Python package to discover association rules in Pandas DataFrames.

Installation

To install the package:

pip install ruleminer

To install the package from Github:

pip install -e git+https://github.com/wjwwillemse/ruleminer.git#egg=ruleminer

To use ruleminer in a project:

import ruleminer

Usage

Suppose we have the following data.

Some insurance undertakings data

Name

Type

Assets

TP-life

TP-nonlife

Own funds

Excess

Insurer 1

life insurer

1000

800

0

200

200

Insurer 2

non-life insurer

4000

0

3200

800

800

Insurer 3

non-life insurer

800

0

700

100

100

Insurer 4

life insurer

2500

1800

0

700

700

Insurer 5

non-life insurer

2100

0

2200

200

200

Insurer 6

life insurer

9000

8800

0

200

200

Insurer 7

non-life insurer

9000

0

8800

200

200

Insurer 8

life insurer

9000

8800

0

200

200

Insurer 9

non-life insurer

9000

8800

0

200

200

Insurer 10

life insurer

9000

0

8800

200

199.99

Calculating metrics

Take for example the rule:

if ({"Type"} == "life_insurer") then ({"TP-life"} > 0)

With the code:

templates = [{'expression': 'if ({"Type"} == "life_insurer") then ({"TP-life"} > 0)'}]

r = ruleminer.RuleMiner(templates=templates, data=df)

you can generate the rule metrics for this rule given the data in the DataFrame above (available with r.rules).

Generated rules (1)

id

definition

status

abs support

abs exceptions

confidence

encodings

1

if ({“Type”} == “life_insurer”) then ({“TP-life”} > 0)

None

5

0

1

{}

There are 5 lines in the data that support this rule. There are no exceptions (i.e. where the if-clause is satisfied, but not the then-clause), so this rule has confidence 1.

Generating rules

You can define rule templates that contain regular expressions for column names and strings. The package will then generate rules that satisfy the rule template with matching column names and strings from the DataFrame. For example, given the data DataFrame above, column regex:

{"T.*"}

will satisfy column names:

{"Type"}, {"TP-life"}, {"TP-nonlife"}

So, if you apply the following rule

if ({"T.*"} == ".*") then ({"TP.*"} > 0)

then the following rules are generated

Generated rules (2)

id

definition

status

abs support

abs exceptions

confidence

encodings

0

if ({“Type”} == “non-life_insurer”) then ({“TP-nonlife”} > 0)

None

4

1

0.8

{}

1

if ({“Type”} == “life_insurer”) then ({“TP-life”} > 0)

None

5

0

1

{}

You can use rules without an if-clause, for example:

{"Assets"} > 0

The metrics for these rules are calculated as if the if-clause is always satisfied.

Parameters

Rule metrics

Several rule metrics have been proposed in the past. You can add the metrics that you want as a parameter to the ruleminer, i.e.:

params = {'metrics': ['added value', 'abs support', 'abs exceptions', 'confidence']}

r = ruleminer.RuleMiner(templates=templates, data=df, params=params)

This will produce the desired metrics. Available metrics are: * abs support * abs exceptions * confidence * support * added value * casual confidence * casual support * conviction * lift

The default metrics are ‘abs support’, ‘abs exceptions’ and ‘confidence’.

See for the definitions Measures for Rules from Michael Hahsler.

Rule precision

In many situations the equal-operator when used on quantitative data is too strict as small differences can occur that you do no want to consider as exceptions to the rule. For this you can define a decimal parameter inside the params dictionary by this

params={'decimal': 3}

This means that comparisons like:

A==B

are translated to

abs(A-B) <= 1.5*10**(-decimal)

If no ‘decimal’ parameter is provided then the absolute difference should be exactly zero.

Rule pruning

By using regex in column names, it will sometimes happen that rules are identical to other rules, except that they have a different ordering of columns. For example:

max({"TP life"}, {"TP nonlife"})

is identical to:

max({"TP nonlife"}, {"TP life"})

The generated rules are therefore pruned to delete the identical rules from the generated list of rules.

  • a==b is identical to b==a

  • a!=b is identical to b!=a

  • min(a, b) is identical to min(b, a)

  • max(a, b) is identical to max(b, a)

  • a+b is identical to b+a

  • a*b is identical to b*a

These identities are applied recursively in rules. So the rule:

(({"4"}>{"3"}) & (({"2"}+{"1"})=={"0"}))

is identical to:

((({"1"}+{"2"})=={"0"}) & ({"4"}>{"3"}))

and will therefore be pruned from the list if the latter rule is already in the list.

Rule template grammar

The rule template describes the structure of the rule. Columns and quoted strings in the rule template can contain simple regular expressions.

Examples:

{"Assets"} > 0

if ({"Type"} == "life insurer") then ({".*"} > 0)

if ({".*"} > 0) then (({".*"} == 0) & ({".*"} > 0))

(({".*"} + {".*"} + {".*"}) == {".*"})

(min({".*"}, {".*"}) == {".*"})

The syntax of the template follows a grammar defined as follows:

  • a template is of the form:

    if cond_1 then cond_2

    or simply a single:

    cond_1
  • a condition is either a combination of comparisons with logical operators (’&’ and ‘|’) and parenthesis:

    ( comp_1 & comp_2 | comp_3 )

    or simply a single comparison:

    comp_1
  • a comparison consists of a term, a comparison operator (>=, >, <=, <, != or ==) and a term, so:

    term_1 > term_2
  • a term can be a number (e.g. 3.1415 or 9), quoted string (a string with single or double quotes), or a function of columns

  • a function of columns is either a prefix operator (min, max or abs, in lower or uppercase) on one or more columns, and of the form, for example:

    min(col_1, col_2, col_3)

    or infix operators with one or more columns:

    (col_1 + col_2 * col_3)
  • a column is a string with braces, so:

    {"Type"}

    where “Type” is the name of the column in the DataFrame with the data

  • a string consists of a-z A-Z 0-9 _ . , ; ; < > * = + - / ? | @ # $ % ^ & ( )

Debugging rules

If you are using this in a Jupyter notebook you can add a the beginning:

logging.basicConfig(stream=sys.stdout,
                    format='%(asctime)s %(message)s',
                    level=logging.INFO)

Info about the rule generating process with be displayed in the notebook.

History

0.1.0 (2021-11-21)

  • First release on PyPI.

0.1.1 (2021-11-23)

  • Added more documentation to the README text

0.1.2 (2022-1-20)

  • Bug fixes wrt some complex expressions

0.1.3 (2022-1-26)

  • Optimized rule generation process

0.1.4 (2022-1-26)

  • Evaluated columns in then part are now dependent on if part of rule

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

ruleminer-0.1.4-py2.py3-none-any.whl (15.4 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page