blacklight/micmon
1
0
Fork 0
mirror of https://github.com/BlackLight/micmon.git synced 2024-12-27 11:05:12 +01:00
Code Issues Projects Releases Packages Wiki Activity

More Keras back-compatibility fixes

Browse source
This commit is contained in:
Fabio Manganiello 2020-10-28 19:46:20 +01:00
parent 870f7ad6b4
commit 604c315b5e
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
micmon/model/__init__.py
View file

@ -2,8 +2,8 @@ import json
import os
import numpy as np
from typing import List, Optional, Union
from keras import Sequential, losses, optimizers, metrics
from typing import List, Optional
from keras import Sequential
from keras.layers import Layer
from keras.models import load_model, Model as _Model
@ -17,9 +17,10 @@ class Model:
# noinspection PyShadowingNames
def __init__(self, layers: Optional[List[Layer]] = None, labels: Optional[List[str]] = None,
model: Optional[_Model] = None, optimizer: Union[str, optimizers.Optimizer] = 'adam',
loss: str = 'sparse_categorical_crossentropy',
metrics: List[Union[str, metrics.Metric]] = ('accuracy',),
model: Optional[_Model] = None,
optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=('accuracy',),
low_freq: int = AudioSegment.default_low_freq,
high_freq: int = AudioSegment.default_high_freq):
assert layers or model