optimizer_type.h

Go to the documentation of this file.

#ifndef OPTIMIZER_TYPE_H
#define OPTIMIZER_TYPE_H
 
#include "cuberl/extern/enum.h"
#include <string>
 
namespace cuberl {
namespace maths{
namespace optim{
 
BETTER_ENUM(OptimzerType, char, INVALID_TYPE=-1, GD=0, SGD, ADAM, RSPROP);
 
 
inline
std::string to_string(OptimzerType type){return type._to_string();}
 
inline
constexpr bool compare(OptimzerType tp1, OptimzerType tp2){
    return tp1 == tp2;
}
 
}
 
}
}
#endif // OPTIMIZER_TYPE_H