728x90 반응형 tensorflow1 Tensorflow.js 선형회귀분석 모델 생성과 테스트하기 1. 모델 생성까지의 과정 let X = tf.input({ shape: [1] }); //독립변수 개수 let Y = tf.layers.dense({ units: 1 }).apply(X); //출력될 결과(종속)변수 개수 let model = tf.model({ inputs: X, outputs: Y }); //모델 정의 let compileParam = { optimizer: tf.train.adam()/*방식*/, loss: tf.losses.meanSquaredError/*손실함수*/ } model.compile(compileParam); //모델 컴파일링 let fitParam = { epochs: 2000, //학습횟수 callbacks:{ onEpochEnd: function(epoch, l.. 2021. 4. 17. 이전 1 다음 728x90 반응형