Skip to content

Commit

Permalink
docs(stacking): fix example
Browse files Browse the repository at this point in the history
  • Loading branch information
SherlockMones committed Aug 29, 2022
1 parent debc65f commit f555fce
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions aucmedi/ensemble/stacking.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,7 @@ class Stacking:
# For standardize_mode
model_b = NeuralNetwork(n_labels=4, channels=3, architecture="2D.MobileNetV2")
model_b.meta_standardize = "torch"
```
??? warning "NeuralNetwork re-initialization"
The passed NeuralNetwork for the train() and predict() function of the Composite class will be re-initialized!
Expand Down

0 comments on commit f555fce

Please sign in to comment.