darknet  v3
Classes | Functions
rnn.c File Reference
#include "darknet.h"
#include <math.h>
Include dependency graph for rnn.c:

Go to the source code of this file.

Classes

struct  float_pair
 

Functions

unsigned char ** load_files (char *filename, int *n)
 
int * read_tokenized_data (char *filename, size_t *read)
 
char ** read_tokens (char *filename, size_t *read)
 
float_pair get_rnn_token_data (int *tokens, size_t *offsets, int characters, size_t len, int batch, int steps)
 
float_pair get_seq2seq_data (char **source, char **dest, int n, int characters, size_t len, int batch, int steps)
 
float_pair get_rnn_data (unsigned char *text, size_t *offsets, int characters, size_t len, int batch, int steps)
 
void train_char_rnn (char *cfgfile, char *weightfile, char *filename, int clear, int tokenized)
 
void print_symbol (int n, char **tokens)
 
void test_char_rnn (char *cfgfile, char *weightfile, int num, char *seed, float temp, int rseed, char *token_file)
 
void test_tactic_rnn_multi (char *cfgfile, char *weightfile, int num, float temp, int rseed, char *token_file)
 
void test_tactic_rnn (char *cfgfile, char *weightfile, int num, float temp, int rseed, char *token_file)
 
void valid_tactic_rnn (char *cfgfile, char *weightfile, char *seed)
 
void valid_char_rnn (char *cfgfile, char *weightfile, char *seed)
 
void vec_char_rnn (char *cfgfile, char *weightfile, char *seed)
 
void run_char_rnn (int argc, char **argv)
 

Function Documentation

◆ get_rnn_data()

float_pair get_rnn_data ( unsigned char *  text,
size_t *  offsets,
int  characters,
size_t  len,
int  batch,
int  steps 
)

Definition at line 127 of file rnn.c.

◆ get_rnn_token_data()

float_pair get_rnn_token_data ( int *  tokens,
size_t *  offsets,
int  characters,
size_t  len,
int  batch,
int  steps 
)

Definition at line 70 of file rnn.c.

◆ get_seq2seq_data()

float_pair get_seq2seq_data ( char **  source,
char **  dest,
int  n,
int  characters,
size_t  len,
int  batch,
int  steps 
)

Definition at line 96 of file rnn.c.

◆ load_files()

unsigned char** load_files ( char *  filename,
int *  n 
)

Definition at line 10 of file rnn.c.

◆ print_symbol()

void print_symbol ( int  n,
char **  tokens 
)

Definition at line 237 of file rnn.c.

◆ read_tokenized_data()

int* read_tokenized_data ( char *  filename,
size_t *  read 
)

Definition at line 24 of file rnn.c.

◆ read_tokens()

char** read_tokens ( char *  filename,
size_t *  read 
)

Definition at line 47 of file rnn.c.

◆ run_char_rnn()

void run_char_rnn ( int  argc,
char **  argv 
)

Definition at line 519 of file rnn.c.

◆ test_char_rnn()

void test_char_rnn ( char *  cfgfile,
char *  weightfile,
int  num,
char *  seed,
float  temp,
int  rseed,
char *  token_file 
)

Definition at line 245 of file rnn.c.

◆ test_tactic_rnn()

void test_tactic_rnn ( char *  cfgfile,
char *  weightfile,
int  num,
float  temp,
int  rseed,
char *  token_file 
)

Definition at line 344 of file rnn.c.

◆ test_tactic_rnn_multi()

void test_tactic_rnn_multi ( char *  cfgfile,
char *  weightfile,
int  num,
float  temp,
int  rseed,
char *  token_file 
)

Definition at line 299 of file rnn.c.

◆ train_char_rnn()

void train_char_rnn ( char *  cfgfile,
char *  weightfile,
char *  filename,
int  clear,
int  tokenized 
)

Definition at line 157 of file rnn.c.

◆ valid_char_rnn()

void valid_char_rnn ( char *  cfgfile,
char *  weightfile,
char *  seed 
)

Definition at line 435 of file rnn.c.

◆ valid_tactic_rnn()

void valid_tactic_rnn ( char *  cfgfile,
char *  weightfile,
char *  seed 
)

Definition at line 386 of file rnn.c.

◆ vec_char_rnn()

void vec_char_rnn ( char *  cfgfile,
char *  weightfile,
char *  seed 
)

Definition at line 473 of file rnn.c.