Skip to content

Commit

Permalink
updated style
Browse files Browse the repository at this point in the history
  • Loading branch information
cnellington committed Feb 17, 2024
1 parent af62b4c commit ed296d5
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
12 changes: 6 additions & 6 deletions contextualized/analysis/pvals.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,7 +169,11 @@ def calc_heterogeneous_predictor_effects_pvals(model, C, **kwargs):


def test_sequential_contexts(
model_constructor: Type[SKLearnWrapper], C: pd.DataFrame, X: pd.DataFrame, Y:pd.DataFrame, **kwargs
model_constructor: Type[SKLearnWrapper],
C: pd.DataFrame,
X: pd.DataFrame,
Y: pd.DataFrame,
**kwargs,
) -> pd.DataFrame:
"""
Sequentially test each feature in C using calc_homogeneous_context_effects_pvals.
Expand All @@ -184,11 +188,7 @@ def test_sequential_contexts(
Returns:
pd.DataFrame: A DataFrame of p-values for each feature.
"""
default_fit_params = {
'encoder_type': 'mlp',
'max_epochs': 3,
'learning_rate': 1e-2
}
default_fit_params = {"encoder_type": "mlp", "max_epochs": 3, "learning_rate": 1e-2}
fit_params = {**default_fit_params, **kwargs}
pvals_dict = {}

Expand Down
2 changes: 1 addition & 1 deletion pylint.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit ed296d5

Please sign in to comment.