Skip to content

Commit de5b15e

Browse files
committed
Update name of project
1 parent 7a459da commit de5b15e

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed
File renamed without changes.

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
long_description = fh.read()
55

66
setuptools.setup(
7-
name="ml_workflow",
7+
name="iaflow",
88
version="1.0.0",
99
author="Enmanuel Magallanes Pinargote",
1010
author_email="enmanuelmag@cardor.dev",

test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22
import tensorflow as tf
3-
import ml_workflow as mlw
3+
import iaflow as iaf
44

55

66
def custom_builder(input_shape):
@@ -13,7 +13,7 @@ def custom_builder(input_shape):
1313
return model
1414

1515

16-
model, run_id, model_ident, check_path, path_model = mlw.build(
16+
model, run_id, model_ident, check_path, path_model = iaf.build(
1717
model_name='model_dense', # Name of the model
1818
models_folder='./models', # Path to folder where models will be saved
1919
run_id=None, # Optional run id to load model
@@ -45,7 +45,7 @@ def custom_builder(input_shape):
4545

4646

4747

48-
mlw.train(
48+
iaf.train(
4949
model=model, # The tf keras model
5050
check_path=check_path, # Path to the model checkpoint
5151
path_model=path_model, # Path to the model folder

0 commit comments

Comments
 (0)