#include <regression.hpp>

Inheritance diagram for jubatus::core::driver::regression:

Collaboration diagram for jubatus::core::driver::regression:

Public Types
typedef core::regression::regression_base	regression_base

Public Member Functions
void	clear ()

float	estimate (const fv_converter::datum &data) const

void	get_status (std::map< std::string, std::string > &status) const

void	pack (framework::packer &pk) const

	regression (jubatus::util::lang::shared_ptr< storage::storage_base > model_storage, jubatus::util::lang::shared_ptr< regression_base > regression_method, jubatus::util::lang::shared_ptr< fv_converter::datum_to_fv_converter > converter)

void	train (const std::pair< float, fv_converter::datum > &data)

void	unpack (msgpack::object o)

virtual	~regression ()

Public Member Functions inherited from jubatus::core::driver::driver_base
	driver_base ()

virtual framework::mixable *	get_mixable ()

std::vector< storage::version >	get_versions () const

virtual	~driver_base ()

Private Attributes
jubatus::util::lang::shared_ptr< fv_converter::datum_to_fv_converter >	converter_

framework::linear_function_mixer	mixable_regression_model_

jubatus::util::lang::shared_ptr< jubatus::core::regression::regression_base >	regression_

fv_converter::mixable_weight_manager	wm_

Additional Inherited Members
Protected Member Functions inherited from jubatus::core::driver::driver_base
void	register_mixable (framework::mixable *mixable)

Protected Attributes inherited from jubatus::core::driver::driver_base
mixable_holder	holder_

Detailed Description

Definition at line 40 of file regression.hpp.

Member Typedef Documentation

typedef core::regression::regression_base jubatus::core::driver::regression::regression_base

Definition at line 42 of file regression.hpp.

Constructor & Destructor Documentation

jubatus::core::driver::regression::regression	(	jubatus::util::lang::shared_ptr< storage::storage_base >	model_storage,
		jubatus::util::lang::shared_ptr< regression_base >	regression_method,
		jubatus::util::lang::shared_ptr< fv_converter::datum_to_fv_converter >	converter
	)

Definition at line 39 of file regression.cpp.

References converter_, jubatus::core::framework::linear_mixable_helper< Model, Diff >::get_model(), mixable_regression_model_, jubatus::core::driver::driver_base::register_mixable(), and wm_.

     : converter_(converter)
     , regression_(regression_method)
     , mixable_regression_model_(regression_method->get_storage())
     , wm_(mixable_weight_manager::model_ptr(new weight_manager)) {
   register_mixable(&mixable_regression_model_);
   register_mixable(&wm_);
 
   converter_->set_weight_manager(wm_.get_model());
 }

Here is the call graph for this function:

jubatus::core::driver::regression::~regression ( )

virtual

Definition at line 53 of file regression.cpp.

53 {

54 }

Member Function Documentation

void jubatus::core::driver::regression::clear ( )

virtual

Implements jubatus::core::driver::driver_base.

Definition at line 74 of file regression.cpp.

References converter_, and regression_.

                        {
   regression_->clear();
   converter_->clear_weights();
 }

float jubatus::core::driver::regression::estimate ( const fv_converter::datum & data ) const

Definition at line 62 of file regression.cpp.

References converter_, regression_, and jubatus::core::common::jsonconfig::v().

                                          {
   common::sfv_t v;
   converter_->convert(data, v);
   float value = regression_->estimate(v);
   return value;
 }

Here is the call graph for this function:

void jubatus::core::driver::regression::get_status ( std::map< std::string, std::string > & status ) const

Definition at line 70 of file regression.cpp.

References regression_.

                                                                 {
   regression_->get_status(status);
 }

void jubatus::core::driver::regression::pack ( framework::packer & pk ) const

virtual

Implements jubatus::core::driver::driver_base.

Definition at line 79 of file regression.cpp.

References jubatus::core::framework::linear_mixable_helper< Model, Diff >::get_model(), regression_, and wm_.

                                                {
   pk.pack_array(2);
   regression_->get_storage()->pack(pk);
   wm_.get_model()->pack(pk);
 }

Here is the call graph for this function:

void jubatus::core::driver::regression::train ( const std::pair< float, fv_converter::datum > & data )

Definition at line 56 of file regression.cpp.

References converter_, regression_, and jubatus::core::common::jsonconfig::v().

                                                                  {
   common::sfv_t v;
   converter_->convert_and_update_weight(data.second, v);
   regression_->train(v, data.first);
 }

Here is the call graph for this function:

void jubatus::core::driver::regression::unpack ( msgpack::object o )

virtual

Implements jubatus::core::driver::driver_base.

Definition at line 85 of file regression.cpp.

References converter_, jubatus::core::framework::linear_mixable_helper< Model, Diff >::get_model(), regression_, and wm_.

                                        {
   if (o.type != msgpack::type::ARRAY || o.via.array.size != 2) {
     throw msgpack::type_error();
   }
 
   // clear before load
   regression_->clear();
   converter_->clear_weights();
   regression_->get_storage()->unpack(o.via.array.ptr[0]);
   wm_.get_model()->unpack(o.via.array.ptr[1]);
 }