implemented NS(n)

1 files changed, 4 insertions, 6 deletions

diff --git a/utils.py b/utils.py
index c5e020f..43f2b4a 100644
--- a/utils.py
+++ b/utils.py
@@ -1,12 +1,10 @@
+from brown import brown_collection
 from consts import SN, SN_
 
-def NS(n):
+def NS(noun):
     """Noun sexiness function
     Returns a real value measure of the maximum similarity a noun `n not in SN`
     has to each of the `nouns in SN_`
     """
-    for noun in SN_:
-        acv = ()
-        # let acv be the vector of absolute frequencies of each adjective that
-        # modifies the noun in the union of the erotica and Brown corpora
-    return 0
+    # return 10**-7 for nouns that occur less than 200 times
+    return brown_collection.tf_idf(noun, SN_)