a91082900/RecSim_FlashcardLearning
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix random seeds with np.randon.RandomState()

Browse Source
This commit is contained in:
Jerry Wu
2023-10-29 23:18:48 +08:00
parent 84a08b08ee
commit a583855e47
5 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
document/FlashcardDocumentSampler.py
View File

@@ -2,13 +2,13 @@ from .FlashcardDocument import FlashcardDocument
from recsim import document
class FlashcardDocumentSampler(document.AbstractDocumentSampler):
def __init__(self, doc_ctor=FlashcardDocument, **kwargs):
super(FlashcardDocumentSampler, self).__init__(doc_ctor, **kwargs)
def __init__(self, doc_ctor=FlashcardDocument, seed=0, **kwargs):
super(FlashcardDocumentSampler, self).__init__(doc_ctor, seed, **kwargs)
self._doc_count = 0
def sample_document(self):
doc_features = {}
doc_features['doc_id'] = self._doc_count
doc_features['difficulty'] = self._rng.uniform(0, 5, (1, 3))
doc_features['difficulty'] = self._rng.uniform(0, 3, (1, 3))
self._doc_count += 1
return self._doc_ctor(**doc_features)