initial check in

1 files changed, 19 insertions, 0 deletions

diff --git a/src/freqdict.h b/src/freqdict.h
new file mode 100644
index 00000000..c9bb4c42
--- /dev/null
+++ b/src/freqdict.h
@@ -0,0 +1,19 @@
+#ifndef _FREQDICT_H_
+#define _FREQDICT_H_
+
+#include <map>
+#include <string>
+
+class FreqDict {
+ public:
+  void load(const std::string& fname);
+  float frequency(const std::string& word) const {
+    std::map<std::string,float>::const_iterator i = counts_.find(word);
+    if (i == counts_.end()) return 0;
+    return i->second;
+  }
+ private:
+  std::map<std::string, float> counts_;
+};
+
+#endif