[TEST] hLouvain

src/Data/Graph/Clustering/Louvain.hs

@@ -38,11 +38,11 @@ type Community = [Node]
 type Reverse = Bool
 
 ------------------------------------------------------------------------
-iLouvain :: I.MaxIterations
+iLouvainMap :: I.MaxIterations
          -> I.MaxSize
          -> Map (Node, Node) Double
          -> [LouvainNode]
-iLouvain x s m = concat
+iLouvainMap x s m = concat
          $ toLouvainNode
          $ I.toNodes
          $ I.iLouvain x s I.DfsNodes g
@@ -55,11 +55,16 @@ flouvain n g = map (comNodes . snd) $ labNodes g'
   where
     g' = louvainFirstStepIterate n (toFGraph g)
 ------------------------------------------------------------------------
-hLouvain :: (Eq b, DynGraph gr)
+
+hLouvain :: Map (Node, Node) Double
+         -> [LouvainNode]
+hLouvain m = hLouvain' False $ undir $ map2graph m
+
+hLouvain' :: (Eq b, DynGraph gr)
          => Reverse
          -> gr a b
          -> [LouvainNode]
-hLouvain r g = concat $ toLouvainNode (bestpartition r g)
+hLouvain' r g = concat $ toLouvainNode (bestpartition r g)
 
 toLouvainNode :: [[Node]] -> [[LouvainNode]]
 toLouvainNode ns = zipWith (\cId ns' -> map (\n -> LouvainNode n cId) ns')