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]; }