change weights for document and user
This commit is contained in:
parent
b4a71cd8e4
commit
5d87afdcc0
@ -9,6 +9,6 @@ class FlashcardDocumentSampler(document.AbstractDocumentSampler):
|
|||||||
def sample_document(self):
|
def sample_document(self):
|
||||||
doc_features = {}
|
doc_features = {}
|
||||||
doc_features['doc_id'] = self._doc_count
|
doc_features['doc_id'] = self._doc_count
|
||||||
doc_features['difficulty'] = self._rng.random_sample((1, 3))
|
doc_features['difficulty'] = self._rng.uniform(0, 5, (1, 3))
|
||||||
self._doc_count += 1
|
self._doc_count += 1
|
||||||
return self._doc_ctor(**doc_features)
|
return self._doc_ctor(**doc_features)
|
@ -52,7 +52,7 @@ class FlashcardUserModel(user.AbstractUserModel):
|
|||||||
doc_id = doc._doc_id
|
doc_id = doc._doc_id
|
||||||
W = self._user_state._W[doc_id]
|
W = self._user_state._W[doc_id]
|
||||||
if not W.any(): # uninitialzed
|
if not W.any(): # uninitialzed
|
||||||
self._user_state._W[doc_id] = W = doc.base_difficulty + np.random.uniform(-0.5, 0.5, (1, 3)) # a uniform error for each user
|
self._user_state._W[doc_id] = W = doc.base_difficulty * np.random.uniform(0.5, 2.0, (1, 3)) # a uniform error for each user
|
||||||
print(W)
|
print(W)
|
||||||
# use exponential function to simulate whether the user recalls
|
# use exponential function to simulate whether the user recalls
|
||||||
last_review = self._user_state._time - self._user_state._last_review[doc_id]
|
last_review = self._user_state._time - self._user_state._last_review[doc_id]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user