30 lines
737 B
Python
30 lines
737 B
Python
import nltk
|
|
import os
|
|
|
|
|
|
def main():
|
|
nltk.download("wordnet")
|
|
|
|
adjectives = {
|
|
lemma.name().capitalize()
|
|
for synset in nltk.corpus.wordnet.all_synsets(pos="a")
|
|
for lemma in synset.lemmas()
|
|
if lemma.name().isalpha()
|
|
}
|
|
nouns = {
|
|
lemma.name().capitalize()
|
|
for synset in nltk.corpus.wordnet.all_synsets(pos="n")
|
|
for lemma in synset.lemmas()
|
|
if lemma.name().isalpha()
|
|
}
|
|
|
|
os.makedirs("data", exist_ok=True)
|
|
with open("data/adjectives.txt", "w") as adj_file:
|
|
adj_file.write("\n".join(sorted(adjectives)))
|
|
with open("data/nouns.txt", "w") as noun_file:
|
|
noun_file.write("\n".join(sorted(nouns)))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|