INNER CODE UNIT · Python
get_train_size
MITDeepLearning/introtodeeplearning · mitdeeplearning/lab2.py:69
def get_train_size(self):
return self.train_inds.shape[0]
def get_train_steps_per_epoch(self, batch_size, factor=10):
return self.get_train_size() // factor // batch_size
def get_batch(self, n, only_faces=False, p_pos=None, p_neg=None, return_inds=False):
if only_faces:
selected_inds = np.random.choice(
self.pos_train_inds, size=n, replace=False, p=p_pos
)
else:
selected_pos_inds = np.random.choice(
self.pos_train_inds, size=n // 2, replace=False, p=p_pos
)
selected_neg_inds = np.random.choice(
self.neg_train_inds, size=n // 2, replace=False, p=p_neg
)