def read_stopwords():
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == StopwordHandler.db_type_fs:
return DiskStorageStopwordHandler.read_stopwords(session_id)
else:
return list()
def add_stopwords(stopwords):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == StopwordHandler.db_type_fs:
DiskStorageStopwordHandler.add_stopwords(session_id, stopwords)
SessionLogger.log(
str(len(stopwords)) + ' stop words have been added.')
def read_categories():
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == CategoryListHandler.db_type_fs:
return DiskStorageCategoryListHandler.read_categories(session_id)
else:
return list()
def add_evaluation(score, session_id=None):
if session_id is None:
session_id = ConfigReader.get_session_id()
config_id = ConfigReader.get_config_id()
evaluation_frame = Storage.load_pd_frame(
EvaluationHandler.evaluations_id, session_id=session_id)
timestamp_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
row = len(evaluation_frame)
evaluation_frame.at[row,
EvaluationHandler.timestamp_col] = timestamp_str
evaluation_frame.at[row, EvaluationHandler.session_id_col] = session_id
evaluation_frame.at[row, EvaluationHandler.config_id_col] = config_id
evaluation_frame.at[row, EvaluationHandler.score_col] = score
Storage.store_pd_frame(evaluation_frame,
EvaluationHandler.evaluations_id,
session_id=session_id)
def add_categories(categories):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == CategoryListHandler.db_type_fs:
DiskStorageCategoryListHandler.add_categories(session_id, categories)
SessionLogger.log(str(len(categories)) + ' categories have been added.')
def delete_session_data():
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
DiskStorage.delete_session_data(session_id)
def load_h5_model(identifier):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
return DiskStorage.load_h5_model(identifier, session_id)
def delete_h5_model(identifier):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
DiskStorage.delete_h5_model(identifier, session_id)
def store_model(model, identifier):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
DiskStorage.store_model(model, identifier, session_id)
def delete_pd_frame(identifier, session_id=None):
db_type = ConfigReader.get_db_type()
if session_id is None:
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
DiskStorage.delete_pd_frame(identifier, session_id)
def load_pd_frame(identifier, session_id=None):
db_type = ConfigReader.get_db_type()
if session_id is None:
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
return DiskStorage.load_pd_frame(identifier, session_id)
def list_ids(location):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
return DiskStorage.list_ids(location, session_id)
def delete_location(location):
db_type = ConfigReader.get_db_type()
session_id = ConfigReader.get_session_id()
if db_type == Storage.db_type_fs:
DiskStorage.delete_location(location, session_id)
