Removed scale factor from second-order LINE

jdr0887 · Aug 9, 2022 · e59e63f · e59e63f
1 parent 32ea00a
commit e59e63f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/graph/cpu_models/src/second_order_line.rs b/graph/cpu_models/src/second_order_line.rs
@@ -49,8 +49,7 @@ impl GraphEmbedder for SecondOrderLINE {
     }
 
     fn _fit_transform(&self, graph: &Graph, embedding: &mut [&mut [f32]]) -> Result<(), String> {
-        let scale_factor = (self.model.embedding_size as f32).sqrt();
-        let mut learning_rate = self.model.learning_rate / scale_factor;
+        let mut learning_rate = self.model.learning_rate;
         let mut random_state = self.get_random_state();
 
         let shared_node_embedding = ThreadDataRaceAware::new(embedding);