feat: add category selection

Safe-DS · Dec 17, 2024 · dc48837 · dc48837
1 parent 57adaf3
commit dc48837
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/DropoutLayer.sdsstub
@@ -8,7 +8,7 @@ from safeds.ml.nn.typing import ModelImageSize
  *
  * @param probability The probability that an input element gets zeroed out.
  */
- @Category(DataScienceCategory.ModelingQNeuralNetwork)
+@Category(DataScienceCategory.ModelingQNeuralNetwork)
 class DropoutLayer(
     probability: Float
 ) sub Layer {

diff --git a/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub b/packages/safe-ds-lang/src/resources/builtins/safeds/ml/nn/layers/MaxPooling2DLayer.sdsstub
@@ -11,6 +11,7 @@ from safeds.ml.nn.typing import ModelImageSize
  * @param padding the padding of the pooling
  */
 @Experimental
+@Category(DataScienceCategory.ModelingQNeuralNetwork)
 class MaxPooling2DLayer(
     @PythonName("kernel_size") kernelSize: Int,
     stride: Int = -1,