---

OfflineCluster.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2002 University of Massachusetts.  All Rights Reserved.
 *
 * Use of the Lemur Toolkit for Language Modeling and Information Retrieval
 * is subject to the terms of the software license set forth in the LICENSE
 * file included with this software, and also available at
 * http://www.lemurproject.org/license.html
 *
 *==========================================================================
*/

// David Fisher
// init: 02/03/2003
#ifndef _OFFLINECLUSTER_HPP
#define _OFFLINECLUSTER_HPP
#include <set>
#include "common_headers.hpp"
#include "Index.hpp"
#include "ClusterParam.hpp"
#include "ClusterFactory.hpp"
#include "SimFactory.hpp"

class OfflineCluster
{
public:
  OfflineCluster(const Index &ind, 
                 enum ClusterParam::simTypes simType = ClusterParam::COS,
                 enum ClusterParam::clusterTypes clusterType = ClusterParam::CENTROID,
                enum ClusterParam::docModes docMode = ClusterParam::DMAX);

  ~OfflineCluster();

  vector<Cluster*> *kMeans(vector<DOCID_T> docIds, int numParts = 2, 
                           int maxIters = 100);

  vector<Cluster*> *kMeans(Cluster *cluster, int numParts = 2, 
                           int maxIters = 100);

  vector<Cluster*> *bisecting_kMeans(vector<DOCID_T> docIds, int numParts = 2, 
                                     int numIters = 5, int maxIters = 100);

private:
  const SimilarityMethod *sim;
  ClusterFactory *factory;
  const Index &index;
  bool compareClusterSets(Cluster **, Cluster **, int n);
  vector <DOCID_T> selectSeeds(vector<DOCID_T> docIds, int num);
  Cluster *chooseSplit(vector<Cluster *> *working);  
  double scoreSet(vector<Cluster *> *working);
};
#endif

---