Create assignment2.py

Author: Arun Kirubarajan

assignment2/assignment2.py

@@ -0,0 +1,43 @@
+import numpy as np
+import pandas as pd
+from collections import defaultdict
+
+
+def scavenger_hunt() -> dict:
+ pass
+
+
+def get_words(file_path) -> list:
+ pass
+
+
+def get_ngrams(words, size) -> list:
+ pass
+
+
+def get_counts(n_grams) -> dict:
+ pass
+
+
+def generate_gram(counts, context) -> str:
+ pass
+
+
+def generate_sentence(counts, context, length=10) -> list:
+ pass
+
+
+def stringify(sentence) -> list:
+ return " ".join(" ".join(gram) for gram in sentence)
+
+
+if __name__ == '__main__':
+ # create model
+ words = get_words('corpus.txt')
+ n_grams = get_ngrams(words, 1)
+ counts = get_counts(n_grams)
+ 
+ # generate text
+ context = n_grams[np.random.choice(len(n_grams))]
+ sentence = generate_sentence(counts, context, length=50)
+ print(stringify(sentence))