16 lines
456 B
Python
16 lines
456 B
Python
from model.test_model import TestModel
|
|
from pathlib import Path
|
|
from dataset.dataset import Dataset
|
|
|
|
if __name__ == "__main__":
|
|
# Loading dataset
|
|
train_dataset = Dataset(Path('data/resized_dataset/train'))
|
|
valid_dataset = Dataset(Path('data/resized_dataset/valid'))
|
|
for i in train_dataset.take(1):
|
|
print(i)
|
|
|
|
# Training model
|
|
model = TestModel()
|
|
history = model.fit()
|
|
model.save("./src/model/test_model_final.keras")
|