36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
from ..util.data_loader import LabelConverter
|
||
|
|
|
||
|
|
class MLSettings:
|
||
|
|
"""
|
||
|
|
Manage model and training settings for easy saving and loading
|
||
|
|
"""
|
||
|
|
def __init__(self,
|
||
|
|
num_features=1,
|
||
|
|
num_layers=1,
|
||
|
|
hidden_size=1,
|
||
|
|
bidirectional=True,
|
||
|
|
transforms=[],
|
||
|
|
num_epochs=10,
|
||
|
|
batch_size=5,
|
||
|
|
labels=LabelConverter([]),
|
||
|
|
):
|
||
|
|
self.num_features = num_features
|
||
|
|
self.num_layers = num_layers
|
||
|
|
self.hidden_size = hidden_size
|
||
|
|
self.num_epochs = num_epochs
|
||
|
|
self.bidirectional = bidirectional
|
||
|
|
self.transforms = transforms
|
||
|
|
self.batch_size = batch_size
|
||
|
|
self.labels = labels
|
||
|
|
|
||
|
|
def get_name(self):
|
||
|
|
"""
|
||
|
|
F = num_features
|
||
|
|
L = num_layers
|
||
|
|
H = hidden_size
|
||
|
|
B = bidirectional
|
||
|
|
T = #transforms
|
||
|
|
E = #epochs
|
||
|
|
"""
|
||
|
|
return f"F{self.num_features}L{self.num_layers}H{self.hidden_size}B{'1' if self.bidirectional else '0'}T{len(self.transforms)}"
|