changes in common part

This commit is contained in:
Kamila Bobkowska 2020-06-04 07:33:58 +00:00
parent 9e608db67c
commit 76ec19cde0

View File

@ -20,6 +20,7 @@ from models.garbageDump import Dump
def createSets(): def createSets():
if not path.exists('ClassificationGarbage/trainSet'):
rootDir = 'ClassificationGarbage' rootDir = 'ClassificationGarbage'
typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic'] typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic']
testRatio = 0.2 testRatio = 0.2