tzrec/protos/models/general_rank_model.proto (12 lines of code) (raw):
syntax = "proto2";
package tzrec.protos;
import "tzrec/protos/module.proto";
import "tzrec/protos/simi.proto";
message RocketLaunching {
optional MLP share_mlp = 1;
required MLP booster_mlp =2;
required MLP light_mlp = 3;
optional bool feature_based_distillation = 5 [default = false];
// COSINE = 0; EUCLID = 1;
optional Similarity feature_distillation_function = 6 [default=COSINE];
}