>source

나는 tensorflow를 처음 사용합니다.

질문: DNNLinearCombinedRegressor 와 같은 통조림 추정기에 커스텀 손실 함수를 추가하는 방법 ?

도움을 주셔서 감사합니다!


  • 답변 # 1

    나는 loss_fn 를 찾지 못했다   DNNLinearCombinedRegressor 에 전달할 인수 . 그러나 맞춤 견적 도구를 정의하는 것은 간단 해 보입니다 :

    DNNLinearCombinedRegressor 의 코드는 좋은 참조입니다 .

    다음 단계가 필요하다고 생각합니다 :

      사용자 정의 loss_fn 정의  (당신이 좋아하는 것)

      RegressionHead 정의 또는 사용  사용자 정의 loss_fn 를 사용하는  위에 정의되어 있습니다.

      사용자 정의 head 를 전달하십시오.  당신의 자신의 Regressor 에 . 물론 자신 만의 model_fn 를 복사/빌드해야 할 수도 있습니다.  또한

관련 자료

  • 이전 타사 jar을 사용하지 않고 로컬에서 Linux 시스템 (Java 코드)으로 파일 전송
  • 다음 java - jar 형식의 jar 내부에 종속성이있는 최종 jar를 갖는 방법은 무엇입니까?