syntax = "proto3";

message Empty {
}

message Embedding {
  string embedding = 1;
}

message TrainingConfig {
  string training_data_filename = 1;
  string validation_data_filename = 2;
  string test_data_filename = 3;
  repeated Embedding embeddings = 4;
  int32 epochs = 5;
  float learning_rate = 6;
  int32 batch_size = 7;
  string model_filename = 8;
}

service TrainingConfiguration {
  rpc startTraining(Empty) returns(TrainingConfig);
}
