JS로 만드는 AI : TensorFlow.js - 7. 모델의 정체 (가중치, bias 구하기)

1. 모델의 정체

y = ax + b에서 주로 a는 가중치 b는 bias라고 한다.

2. 가중치와 bias 구하기

기본적으로 model을 통해 가져온 값은 tensor의 형태이므로 tensor에서 값을 가져오는 법을 배워야한다.

model.getWeight()를 통해 mod변수로 가져온다.

해당 내용의 [0]배열은 weights고 [1]은 bias이기 때문에 해당 값들을 위와같이 정의한다.

텐서의연산 내용을 통해 텐서값을 기본값으로 바꾸는 arraySync()를 이용해본다.

해당 내용의 결과값을 보면 다음과 같다.

확인해보기

모델에 20의 값을 넣었을때 예측 값이 4 0.0424... 인 형태로 나타나고 이를 weights와 bias를 통해 계산하였을때도 거의 유사한 값으로 출력되는 것을 확인 할 수 있다.

출처 : 생활코딩