Skip to content

Commit

Permalink
Remove cupy docs FeatureNeutralizer
Browse files Browse the repository at this point in the history
  • Loading branch information
CarloLepelaars committed Jan 9, 2024
1 parent 89c2562 commit 36f2296
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions docs/postprocessing.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,14 +44,6 @@ neutralizer.fit()
neutralized_predictions = neutralizer.predict(X=predictions, features=feature_data, eras=era_data)
```

### Note
Neutralization can be run on the GPU by setting `cuda=True`. When setting this ensure you have CuPy installed.

```bash
pip install cupy
```


## FeaturePenalizer

`FeaturePenalizer` neutralizes predictions using TensorFlow based on provided feature exposures. It's designed to integrate seamlessly with scikit-learn.
Expand Down

0 comments on commit 36f2296

Please sign in to comment.