Update 4-autograd-model.mdx

Use BinaryCrossEntropy for classification instead of MSE Regression
NumPower · Jun 13, 2024 · fbfee04 · fbfee04
1 parent 1a33838
commit fbfee04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensor/intro/4-autograd-model.mdx b/tensor/intro/4-autograd-model.mdx
@@ -90,8 +90,8 @@ public function forward(Tensor $x, Tensor $y): array
     $x = $x->matmul($this->weights_output_layer) + $this->output_bias;
     $x = activation::sigmoid($x); // Sigmoid Activation
 
-    // Mean Squared Error
-    $loss = loss::MeanSquaredError($x, $y, name: 'loss');
+    // Binary Cross Entropy Loss
+    $loss = loss::BinaryCrossEntropy($x, $y, name: 'loss');
     return [$x, $loss];
 }
 ```