신경망 설치를 위한 파일 구성
Computer Science/DeepLearning

신경망 설치를 위한 파일 구성

728x90
반응형

train.py

1. Dataset 생성

2. model  구성  : add(Dense 등)

3. model  엮기  : compile(loss, optimizer, matrix)

4. 학습(training)  : fit(x_train, y_train, epoch, batch, x_val, y_val)

5. 평가(evaluation)  evaluate(

6. model 저장 및 파일 변환 .pb / .ckpt / .h5 파일

 

 

 


test.py

1. Dataset 생성

2. model  불러오기

3. 예측(test: predict)  :  pridict()

4. 투표(voting)

 

 

 

 

 

728x90
반응형