TBF_FCI.fit#

TBF_FCI.fit(df, drivers)[source]#

Fit the temporal causal discovery algorithm to df.

Parameters:
dfpandas.DataFrame

Input dataframe containing the target column and every driver column.

driversSequence[str]

Iterable of column names to be treated as drivers of the target.

Returns:
TBF_FCI

The fitted instance with internal adjacency structures populated.

Examples

model = TBF_FCI(target="Y", max_lag=2)
model.fit(df, drivers=["A", "B"])