-
Notifications
You must be signed in to change notification settings - Fork 0
/
optimizer.py
41 lines (29 loc) · 957 Bytes
/
optimizer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import numpy
from evaluation import Evaluator
from model import Model
class Optimizer:
"""
The Optimizer class provides the interface that any LP Optimizer should implement.
"""
def __init__(self,
model: Model,
hyperparameters: dict,
verbose: bool = True):
self.model = model #type: Model
self.dataset = self.model.dataset
self.verbose = verbose
# create the evaluator to use between epochs
self.evaluator = Evaluator(self.model)
self.verbose = verbose
def train(self,
train_samples: numpy.array,
save_path: str = None,
evaluate_every:int =-1,
valid_samples:numpy.array = None):
pass
# def epoch(self,
# batch_size: int,
# training_samples: numpy.array):
# pass
# def step_on_batch(self, loss, batch):
# pass