jshbrmn/irt-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add basic error handling

Browse Source
This commit is contained in:
Josh Burman
2021-12-10 19:45:03 +00:00
parent d107b66e4f
commit e0fca68f46
3 changed files with 37 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/lib/irt/item_information_function.py
View File

@ -1,4 +1,7 @@
import logging
from lib.irt.models.three_parameter_logistic import ThreeParameterLogistic
from lib.errors.item_generation_error import ItemGenerationError
class ItemInformationFunction():
def __init__(self, irt_model):
@ -8,10 +11,14 @@ class ItemInformationFunction():
# further detailed on page 161, equation 4 here:
# https://www.ncbi.nlm.nih.gov/pmc/articles/PMC5978482/pdf/10.1177_0146621615613308.pdf
def calculate(self, **kwargs):
if self.model_data.model == '3PL':
p = ThreeParameterLogistic(self.model_data, kwargs).result()
q = 1 - p
return (self.model_data.a_param * q * (p - self.model_data.c_param)**2) / (p * ((1 - self.model_data.c_param)**2))
else:
# potentially error out
return None
try:
if self.model_data.model == '3PL':
p = ThreeParameterLogistic(self.model_data, kwargs).result()
q = 1 - p
return (self.model_data.a_param * q * (p - self.model_data.c_param)**2) / (p * ((1 - self.model_data.c_param)**2))
else:
# potentially error out
raise ItemGenerationError("irt model not supported or provided")
except ZeroDivisionError as error:
logging.error(error)
raise ItemGenerationError("params not well formatted", error.args[0])