Closed y-prudent closed 1 year ago
Before:
dknn = DKNN(output_layer_id=[-2]) dknn.fit(model, ds_fit)
After:
dknn = DKNN() dknn.fit(model, ds_fit, output_layer_id=[-2])
[Update] Also, for improved clarity output_layers_id should be renamed feature_layers_id. The feature extractor would then return separately in a tuple: (1) the features (identified by feature_layers_id) and (2) the logits (output of last layer).
output_layers_id
feature_layers_id
After v2:
dknn = DKNN() dknn.fit(model, ds_fit, feature_layers_id=[-2])
Before:
After:
[Update] Also, for improved clarity
output_layers_id
should be renamedfeature_layers_id
. The feature extractor would then return separately in a tuple: (1) the features (identified byfeature_layers_id
) and (2) the logits (output of last layer).After v2: