1 files changed, 7 insertions, 0 deletions
diff --git a/src/util.h b/src/util.h
index 69ebfbe..707e79b 100644
--- a/src/util.h
+++ b/src/util.h
@@ -12,9 +12,14 @@ struct Configs {
     char *loss;
     char **input_keys, **label_keys;
     size_t n_input_keys, n_label_keys;
+    char **categorical_keys, ***categorical_values;
+    size_t n_categorical_keys, *n_categorical_values;
     char *weights_filepath;
     char *config_filepath;
     bool shuffle;
+    /* preprocessing */
+    char **onehot_keys;
+    size_t n_onehot_keys;
     /* cli cfgs */
     char *file_format;
     char *in_filepath;
@@ -32,6 +37,8 @@ void die(const char *fmt, ...);
 void *ecalloc(size_t nmemb, size_t size);
 void *erealloc(void *ptr, size_t size);
 char *e_strdup(const char *s);
+int util_get_key_index(char *key, char **keys, size_t n_keys);
+int util_argmax(double *values, size_t n_values);
 void util_load_cli(struct Configs *ml, int argc, char *argv[]);
 void util_load_config(struct Configs *ml, char *filepath);
 void util_free_config(struct Configs *ml);