Skip to content

Commit

Permalink
[Dataset] Cleaned code
Browse files Browse the repository at this point in the history
  • Loading branch information
YanSte committed Aug 4, 2023
1 parent c74ab56 commit dd5cac0
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/smartkit/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,6 +206,8 @@ def rescale_dataset(*data, scale=1):
out = [ d[:int(scale*len(d))] for d in data ]
return out[0] if len(out)==1 else out



# ==============================
# TensorFlow
# ==============================
Expand Down
3 changes: 3 additions & 0 deletions src/smartkit/main.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@



# ---- Gobal import ----

import smartkit.config as config
from smartkit.ModelMetrics import ModelMetrics

Expand Down

0 comments on commit dd5cac0

Please sign in to comment.