We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ea10c5d commit 6925d56Copy full SHA for 6925d56
modelmaker/resources/templates/default/package/scripts/train.py
@@ -1,7 +1,7 @@
1
import numpy as np
2
from tensorflow import keras
3
-from default.models import SimpleClassification
4
-from default.utils import generate_square, generate_circle
+from {{ package_name }}.models import SimpleClassification
+from {{ package_name }}.utils import generate_square, generate_circle
5
6
######################################################################
7
# train SimpleClassification
modelmaker/resources/templates/default/setup.py
@@ -6,6 +6,8 @@
version=__version__,
author='',
8
install_requires=[
9
+ 'numpy',
10
+ 'opencv-python'
11
],
12
url='',
13
long_description=open('README.md').read(),
0 commit comments