diff --git a/assignment1/src/main/java/edu/auburn/extraction/FeatureMap.java b/assignment1/src/main/java/edu/auburn/extraction/FeatureMap.java index 9cc3364..be88b1e 100644 --- a/assignment1/src/main/java/edu/auburn/extraction/FeatureMap.java +++ b/assignment1/src/main/java/edu/auburn/extraction/FeatureMap.java @@ -69,4 +69,14 @@ public String toString() { return mapStringBuilder.toString(); } + + public String toStringVector() { + StringBuilder mapStringBuilder = new StringBuilder(); + for (T feature : getFeatures()) { + Integer featureCount = featureDistribution.get(feature); + double featureFraction = ((double) featureCount) / getTotalFeatureCount(); + mapStringBuilder.append(String.format("%f ", featureFraction)); + } + return mapStringBuilder.toString(); +} } diff --git a/assignment1/src/main/java/edu/auburn/extraction/FeatureMapFileWriter.java b/assignment1/src/main/java/edu/auburn/extraction/FeatureMapFileWriter.java index 9ccaf5e..fa3f625 100644 --- a/assignment1/src/main/java/edu/auburn/extraction/FeatureMapFileWriter.java +++ b/assignment1/src/main/java/edu/auburn/extraction/FeatureMapFileWriter.java @@ -20,6 +20,12 @@ public void write(String url, FeatureMap featureMap) throws IOExcepti getFileWriter().write("\n"); } + public void writeVectors(String url, FeatureMap featureMap) throws IOException { + getFileWriter().write("0 0 "); //Dummy Value for TrainingSet's count + getFileWriter().write(featureMap.toStringVector()); + getFileWriter().write("\n"); + } + public void close() throws IOException { getFileWriter().flush(); getFileWriter().close();