fn_kmeans.hpp

Go to the documentation of this file.

// Copyright (C) 2011 the authors listed below
// http://ecocpak.sourceforge.net
// 
// Authors:
// - Dimitrios Bouzas (bouzas at ieee dot org)
// - Nikolaos Arvanitopoulos (niarvani at ieee dot org)
// - Anastasios Tefas (tefas at aiia dot csd dot auth dot gr)
// 
// This file is part of the ECOC PAK C++ library. It is 
// provided without any warranty of fitness for any purpose.
//
// You can redistribute this file and/or modify it under 
// the terms of the GNU Lesser General Public License (LGPL) 
// as published by the Free Software Foundation, either 
// version 3 of the License or (at your option) any later 
// version.
// (see http://www.opensource.org/licenses for more info)





template<typename T1>
inline
void
kmeans
  (
                            Col<u32>&    indices_out,
                            Col<u32>&    ranks_out,
         Mat<typename T1::elem_type>&    centroids_out,
  const Base<typename T1::elem_type,T1>& X,
  const u32                              k, 
  const typename T1::pod_type            lr = 0.01
  )
  {
  arma_extra_debug_sigprint();
  
  arma_check(k == 0, "kmeans(): Number of clusters can't be 0");  

  typedef typename T1::elem_type eT;
  
  const unwrap<T1>   tmp(X.get_ref());
  const Mat<eT>& A = tmp.M;
  
  op_kmeans::direct_kmeans(indices_out, ranks_out, centroids_out, A, k, lr); 
  }



mat
subClasses
  (
  ucolvec& inds, 
  ucolvec& ranks, 
  const mat& samples
  )
  {
  mat centroids;
  ranks = zeros<ucolvec>(2);
  kmeans(inds, ranks, centroids, samples, 2);
  
  return centroids;
  }