ML functions
 
Loading...
Searching...
No Matches
NNBuilder.h File Reference

Implementation of a neural network builder for constructing and registering neural network layers. More...

#include "velox/ml_functions/functions.h"

Go to the source code of this file.

Classes

class  NNBuilder
 Implements a neural network builder for constructing and registering neural network layers. More...
 

Enumerations

enum  Activation { RELU , SOFTMAX , NONE }
 Enumeration of activation functions supported by the neural network builder. More...
 

Detailed Description

Implementation of a neural network builder for constructing and registering neural network layers.

Enumeration Type Documentation

◆ Activation

enum Activation

Enumeration of activation functions supported by the neural network builder.

Enumerator
RELU 

Rectified Linear Unit (ReLU) activation.

SOFTMAX 

Softmax activation.

NONE 

No activation.