Skip to content
Snippets Groups Projects
Commit d53ba529 authored by shruti-khule's avatar shruti-khule
Browse files

Update requirements

parent 75174efd
No related branches found
No related tags found
No related merge requests found
from torch_geometric.data import HeteroData
from nltk.corpus import stopwords
import nltk
# Ensure NLTK stopwords are available
try:
STOP_WORDS = set(stopwords.words('english'))
except LookupError:
nltk.download('stopwords')
STOP_WORDS = set(stopwords.words('english'))
import ssl
class Base:
def __init__(self, path, bag_of_words_size=100, remove_all_false_values=True):
self.path = path
self.bag_of_words_size = bag_of_words_size
self.remove_all_false_values = remove_all_false_values
try:
self.stop_words = set(stopwords.words('english'))
except LookupError:
try:
_create_unverified_https_context = ssl._create_unverified_context
except AttributeError:
pass
else:
ssl._create_default_https_context = _create_unverified_https_context
nltk.download('stopwords')
self.stop_words = set(stopwords.words('english'))
self.dataset = HeteroData()
def fetch_themes(self, num_groups, grouped_keywords_path=''):
......
......@@ -33,12 +33,6 @@ class DBLP(Base):
Each label (genre) gets its own top bag_of_words_size keywords.
"""
try:
self.stop_words = set(stopwords.words('english'))
except LookupError:
nltk.download('stopwords')
self.top_words = set(stopwords.words('english'))
# Load authors
author_ids, author_labels, author_id_dict = self._get_authors()
......
......@@ -431,7 +431,7 @@ def convert_to_owl(data: Union[HeteroData, Data], namespace: str, owlGraphPath:
if isinstance(data, HeteroData):
return SingleGraphOWLConverter(data, namespace, owlGraphPath, high_level_concepts, **kwargs)
else:
return MultiGraphOWLConverter(data, namespace, owlGraphPath, high_level_concepts, **kwargs)
return MultiGraphOWLConverter(data, namespace, owlGraphPath, high_level_concepts=high_level_concepts, **kwargs)
if __name__ == "__main__":
# Example usage with MUTAG dataset
......
......@@ -13,7 +13,10 @@ joblib==1.4.2
mpmath==1.3.0
multidict==6.0.5
networkx==3.3
nltk==3.9.1
numpy==2.1.0
ontolearn==0.6.0
openai==1.59.7
psutil==6.0.0
pyparsing==3.1.2
requests==2.32.3
......@@ -26,4 +29,4 @@ torch_geometric==2.5.3
tqdm==4.66.5
urllib3==2.2.2
yarl==1.9.4
dotenv==0.0.5
\ No newline at end of file
python-dotenv==1.0.1
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment