types.h

Go to the documentation of this file.

#ifndef TYPES_H
#define TYPES_H
 
#pragma once
 
#include <string>
 
namespace txeo {
 
struct DeviceInfo {
    std::string name{};
 
    std::string device_type{};
 
    size_t memory_limit{};
};
 
enum class NormalizationType { MIN_MAX, Z_SCORE };
 
enum class LossFunc { MSE, MAE, MSLE, LCHE };
 
} // namespace txeo
 
#endif