00001 /* 00002 * This program is free software; you can redistribute it and/or modify 00003 * it under the terms of the GNU General Public License as published by 00004 * the Free Software Foundation; either version 3 of the License, or 00005 * (at your option) any later version. 00006 * 00007 * Written (W) 1999-2008 Soeren Sonnenburg 00008 * Copyright (C) 1999-2008 Fraunhofer Institute FIRST and Max-Planck-Society 00009 */ 00010 00011 #ifndef _SPARSELINEARCLASSIFIER_H__ 00012 #define _SPARSELINEARCLASSIFIER_H__ 00013 00014 #include "lib/common.h" 00015 #include "features/SparseFeatures.h" 00016 #include "classifier/Classifier.h" 00017 00019 class CSparseLinearClassifier : public CClassifier 00020 { 00021 public: 00023 CSparseLinearClassifier(); 00024 virtual ~CSparseLinearClassifier(); 00025 00031 virtual CLabels* classify(CLabels* output=NULL); 00032 00034 virtual inline float64_t classify_example(int32_t vec_idx) 00035 { 00036 return features->dense_dot(1.0, vec_idx, w, w_dim, bias); 00037 } 00038 00044 inline void get_w(float64_t** dst_w, int32_t* dst_dims) 00045 { 00046 ASSERT(dst_w && dst_dims); 00047 ASSERT(w && w_dim>0); 00048 *dst_dims=w_dim; 00049 *dst_w=(float64_t*) malloc(sizeof(float64_t) * (*dst_dims)); 00050 ASSERT(*dst_w); 00051 memcpy(*dst_w, w, sizeof(float64_t) * (*dst_dims)); 00052 } 00053 00059 inline void set_w(float64_t* src_w, int32_t src_w_dim) 00060 { 00061 w=src_w; 00062 w_dim=src_w_dim; 00063 } 00064 00069 inline void set_bias(float64_t b) 00070 { 00071 bias=b; 00072 } 00073 00078 inline float64_t get_bias() 00079 { 00080 return bias; 00081 } 00082 00087 inline void set_features(CSparseFeatures<float64_t>* feat) 00088 { 00089 SG_UNREF(features); 00090 SG_REF(feat); 00091 features=feat; 00092 } 00093 00098 inline CSparseFeatures<float64_t>* get_features() { SG_REF(features); return features; } 00099 00100 protected: 00102 int32_t w_dim; 00104 float64_t* w; 00106 float64_t bias; 00108 CSparseFeatures<float64_t>* features; 00109 }; 00110 #endif