docs: add mod titles

src/initializer.rs

@@ -1,3 +1,5 @@
+//! Initializations for weight matrices.
+
 use rand::Rng;
 
 use crate::tensor::Tensor2D;

src/logistic_regression.rs

@@ -1,3 +1,5 @@
+//! Logistic regression model.
+
 use rand::Rng;
 
 pub struct LogisticRegression {

src/loss.rs

@@ -1,3 +1,5 @@
+//! Loss functions.
+
 use crate::Tensor;
 
 /// Mean squared error loss.

src/optimizer.rs

@@ -1,4 +1,4 @@
-//! Implements different optimizers to be used with neural networks.
+//! Optimization algorithms.
 //!
 //! Use to minimize the loss function during the training process of a neural network by
 //! adjusting the weights of the network based on the gradients of the loss function with

src/prelude.rs

@@ -1 +1,5 @@
+//! Engram prelude.
+//!
+//! Contains all the most commonly used types and functions.
+
 pub use crate::{tensor, Activation, Initializer, Network, Optimizer, Tensor};

src/regularization.rs

@@ -1,4 +1,5 @@
-//! This module provides implementations for L1 and L2 regularization.
+//! L1/L2 regularization.
+//!
 //! Regularization can be used to prevent overfitting.
 //!
 //! # Example

src/stats.rs

@@ -1,4 +1,4 @@
-//! This module contains functions for calculating statistics.
+//! Statistics.
 
 use std::{
     collections::HashMap,