bandit.cpp

Go to the documentation of this file.

// Jubatus: Online machine learning framework for distributed environment
// Copyright (C) 2015 Preferred Networks and Nippon Telegraph and Telephone Corporation.
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License version 2.1 as published by the Free Software Foundation.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

#include "bandit.hpp"

#include <map>
#include <string>
#include <utility>
#include <vector>

#include "../bandit/bandit_factory.hpp"

using jubatus::util::lang::shared_ptr;
using jubatus::core::bandit::bandit_factory;

namespace jubatus {
namespace core {
namespace driver {

bandit::bandit(const std::string& method_name,
               const common::jsonconfig::config& param)
    : bandit_(bandit_factory::create(method_name, param)),
      mixable_storage_(bandit_) {
  register_mixable(&mixable_storage_);
}

bool bandit::register_arm(const std::string& arm_id) {
  return bandit_->register_arm(arm_id);
}
bool bandit::delete_arm(const std::string& arm_id) {
  return bandit_->delete_arm(arm_id);
}

std::string bandit::select_arm(const std::string& player_id) {
  return bandit_->select_arm(player_id);
}

bool bandit::register_reward(const std::string& player_id,
                             const std::string& arm_id,
                             double reward) {
  return bandit_->register_reward(player_id, arm_id, reward);
}

core::bandit::arm_info_map bandit::get_arm_info(
      const std::string& player_id) const {
  return bandit_->get_arm_info(player_id);
}

bool bandit::reset(const std::string& player_id) {
  return bandit_->reset(player_id);
}

void bandit::clear() {
  bandit_->clear();
}
void bandit::pack(framework::packer& pk) const {
  bandit_->pack(pk);
}
void bandit::unpack(msgpack::object o) {
  bandit_->unpack(o);
}

}  // namespace driver
}  // namespace core
}  // namespace jubatus