loss_function_factory.h

Go to the documentation of this file.

#ifndef META_CLASSIFY_LOSS_FUNCTION_FACTORY_H_
#define META_CLASSIFY_LOSS_FUNCTION_FACTORY_H_

#include <functional>
#include <memory>
#include <unordered_map>

#include "classify/loss/loss_function.h"
#include "util/factory.h"
#include "util/shim.h"

namespace meta
{
namespace classify
{
namespace loss
{

class loss_function_factory
    : public util::factory<loss_function_factory, loss_function>
{
    friend base_factory;

  private:
    loss_function_factory();

    template <class Loss>
    void reg();
};

std::unique_ptr<loss_function>
    make_loss_function(const std::string& identifier);

template <class Loss>
std::unique_ptr<loss_function> make_loss_function()
{
    return make_unique<Loss>();
}

template <class Loss>
void register_loss_function()
{
    loss_function_factory::get().add(Loss::id, make_loss_function<Loss>);
}
}
}
}

#endif