Attempt at fixing sparse features for hgmira

1 files changed, 3 insertions, 3 deletions

diff --git a/mert/MiraWeightVector.h b/mert/MiraWeightVector.h
index bbc28704b..8200d6013 100644
--- a/mert/MiraWeightVector.h
+++ b/mert/MiraWeightVector.h
@@ -64,9 +64,9 @@ public:
   AvgWeightVector avg();
 
   /**
-    * Convert to sparse vector, interpreting all features as sparse.
+    * Convert to sparse vector, interpreting all features as sparse. Only used by hgmira.
    **/
-  void ToSparse(SparseVector* sparse) const;
+  void ToSparse(SparseVector* sparse, size_t denseSize) const;
 
   friend class AvgWeightVector;
 
@@ -104,7 +104,7 @@ public:
   ValType score(const MiraFeatureVector& fv) const;
   ValType weight(std::size_t index) const;
   std::size_t size() const;
-  void ToSparse(SparseVector* sparse) const;
+  void ToSparse(SparseVector* sparse, size_t num_dense) const;
 private:
   const MiraWeightVector& m_wv;
 };