Feature types¶
Featuretools groups features into four general types:
Identity Features¶
In Featuretools, each feature is defined as a combination of other features. At the lowest level are IdentityFeature
features which are equal to the value of a single variable.
Most of the time, identity features will be defined transparently for you, such as in the transform feature example below. They may also be defined explicitly:
In [1]: time_feature = ft.Feature(es["transactions"]["transaction_time"])
In [2]: time_feature
Out[2]: <Feature: transaction_time>
Direct Features¶
Direct features are used to “inherit” feature values from a parent to a child entity. Suppose each event is associated with a single instance of the entity products.
This entity has metadata about different products, such as brand, price, etc. We can pull the brand of the product into a feature of the event entity by including the event entity as an argument to Feature
.
In this case, Feature
is an alias for primitives.DirectFeature
:
In [3]: brand = ft.Feature(es["products"]["brand"], entity=es["transactions"])
In [4]: brand
Out[4]: <Feature: products.brand>
Transform Features¶
Transform features take one or more features on an Entity
and create a single new feature for that same entity. For example, we may want to take a fine-grained “timestamp” feature and convert it into the hour of the day in which it occurred.
In [5]: from featuretools.primitives import Hour
In [6]: ft.Feature(time_feature, primitive=Hour)
Out[6]: <Feature: HOUR(transaction_time)>
Using algebraic and boolean operations, transform features can combine other features into arbitrary expressions. For example, to determine if a given event event happened in the afternoon, we can write:
In [7]: hour_feature = ft.Feature(time_feature, primitive=Hour)
In [8]: after_twelve = hour_feature > 12
In [9]: after_twelve
Out[9]: <Feature: HOUR(transaction_time) > 12>
In [10]: at_twelve = hour_feature == 12
In [11]: before_five = hour_feature <= 17
In [12]: is_afternoon = after_twelve & before_five
In [13]: is_afternoon
Out[13]: <Feature: AND(HOUR(transaction_time) > 12, HOUR(transaction_time) <= 17)>
Aggregation Features¶
Aggregation features are used to create features for a parent entity by summarizing data from a child entity. For example, we can create a Count
feature which counts the total number of events for each customer:
In [14]: from featuretools.primitives import Count
In [15]: total_events = ft.Feature(es["transactions"]["transaction_id"], parent_entity=es["customers"], primitive=Count)
In [16]: fm = ft.calculate_feature_matrix([total_events], es)
In [17]: fm.head()
Out[17]:
COUNT(transactions)
customer_id
5 79
4 109
1 126
3 93
2 93
Note
For users who have written aggregations in SQL, this concept will be familiar. One key difference in featuretools is that GROUP BY
and JOIN
are implicit. Since the parent and child entities are specified, featuretools can infer how to group the child entity and then join the resulting aggregation back to the parent entity.
Often times, we only want to aggregate using a certain amount of previous data. For example, we might only want to count events from the past 30 days. In this case, we can provide the use_previous
parameter:
In [18]: total_events_last_30_days = ft.Feature(es["transactions"]["transaction_id"],
....: parent_entity=es["customers"],
....: use_previous="30 days",
....: primitive=Count)
....:
In [19]: fm = ft.calculate_feature_matrix([total_events_last_30_days], es)
In [20]: fm.head()
Out[20]:
COUNT(transactions, Last 30 Days)
customer_id
5 0.0
4 0.0
1 0.0
3 0.0
2 0.0
Unlike with cumulative transform features, the use_previous
parameter here is evaluated relative to instances of the parent entity, not the child entity. The above feature translates roughly to the following: “For each customer, count the events which occurred in the 30 days preceding the customer’s timestamp.”
Find the list of the supported aggregation features here.
Where clauses¶
When defining aggregation or cumulative transform features, we can provide a where
parameter to filter the instances we are aggregating over. Using the is_afternoon
feature from earlier, we can count the total number of events which occurred in the afternoon:
In [21]: afternoon_events = ft.Feature(es["transactions"]["transaction_id"],
....: parent_entity=es["customers"],
....: where=is_afternoon,
....: primitive=Count).rename("afternoon_events")
....:
In [22]: fm = ft.calculate_feature_matrix([afternoon_events], es)
In [23]: fm.head()
Out[23]:
afternoon_events
customer_id
5 0.0
4 0.0
1 0.0
3 0.0
2 0.0
The where argument can be any previously-defined boolean feature. Only instances for which the where feature is True are included in the final calculation.
Aggregations of Direct Feature¶
Composing multiple feature types is an extremely powerful abstraction that Featuretools makes simple. For instance, we can aggregate direct features on a child entity from a different parent entity. For example, to calculate the most common brand a customer interacted with:
In [24]: from featuretools.primitives import Mode
In [25]: brand = ft.Feature(es["products"]["brand"], entity=es["transactions"])
In [26]: favorite_brand = ft.Feature(brand, parent_entity=es["customers"], primitive=Mode)
In [27]: fm = ft.calculate_feature_matrix([favorite_brand], es)
In [28]: fm.head()
Out[28]:
MODE(transactions.products.brand)
customer_id
5 B
4 B
1 B
3 B
2 B
Side note: Feature equality overrides default equality¶
Because we can check if two features are equal (or a feature is equal to a value), we override Python’s equals (==) operator. This means to check if two feature objects are equal (instead of their computed values in the feature matrix), we need to compare their hashes:
In [29]: hour_feature.hash() == hour_feature.hash()
Out[29]: True
In [30]: hour_feature.hash() != hour_feature.hash()
Out[30]: False
dictionaries and sets use equality underneath, so those keys need to be hashes as well
In [31]: myset = set()
In [32]: myset.add(hour_feature.hash())
In [33]: hour_feature.hash() in myset
Out[33]: True
In [34]: mydict = dict()
In [35]: mydict[hour_feature.hash()] = hour_feature
In [36]: hour_feature.hash() in mydict
Out[36]: True