first model
This commit is contained in:
@@ -118,8 +118,6 @@ struct model{
|
||||
*/
|
||||
bool add(layer* new_layer);
|
||||
|
||||
// Create and add a dense layer.
|
||||
|
||||
/**
|
||||
* @brief Adds a dense layer to the model.
|
||||
*
|
||||
@@ -140,6 +138,21 @@ struct model{
|
||||
panic::types::uint_t neuron_count
|
||||
);
|
||||
|
||||
/**
|
||||
* @brief Adds a activation ReLU layer to the model.
|
||||
*
|
||||
* Computes:
|
||||
* @code
|
||||
* model.add_activation_ReLU(3,4);
|
||||
* @endcode
|
||||
*
|
||||
*
|
||||
* @return true if layer is added
|
||||
*
|
||||
* @note This function is convenient, but it allocates a new layer.
|
||||
*/
|
||||
bool add_activation_ReLU();
|
||||
|
||||
/**
|
||||
* @brief Loops over all layers forward function
|
||||
*
|
||||
|
||||
Reference in New Issue
Block a user