gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
import * as tf from '../../dist/tfjs.esm';
 
export function seperateWeightMaps(weightMap: tf.NamedTensorMap) {
  const featureExtractorMap: tf.NamedTensorMap = {};
  const classifierMap: tf.NamedTensorMap = {};
 
  Object.keys(weightMap).forEach((key) => {
    const map = key.startsWith('fc') ? classifierMap : featureExtractorMap;
    map[key] = weightMap[key];
  });
 
  return { featureExtractorMap, classifierMap };
}