我正在使用 FANN 来使用神经网络。(链接到 FANN)
我需要在训练网络后得到权重矩阵,但我没有从文档中找到任何东西。(链接到文档)
你知道怎么得到那个矩阵吗???
谢谢!
你需要使用fann_get_connection_array()
函数。它为您提供数组struct fann_connection
,并struct fann_connection
具有字段weight
,所以这就是您想要的。
您可以执行以下操作来打印您的权重矩阵:
int main(void)
{
struct fann *net; /* your trained neural network */
struct fann_connection *con; /* weight matrix */
unsigned int connum; /* connections number */
size_t i;
/* Insert your net allocation and training code here */
...
connum = fann_get_total_connections(net);
if (connum == 0) {
fprintf(stderr, "Error: connections count is 0\n");
return EXIT_FAILURE;
}
con = calloc(connum, sizeof(*con));
if (con == NULL) {
fprintf(stderr, "Error: unable to allocate memory\n");
return EXIT_FAILURE;
}
/* Get weight matrix */
fann_get_connection_array(net, con);
/* Print weight matrix */
for (i = 0; i < connum; ++i) {
printf("weight from %u to %u: %f\n", con[i].from_neuron,
con[i].to_neuron, con[i].weight);
}
free(con);
return EXIT_SUCCESS;
}
细节:
[1] fann_get_connection_array()
[3] fann_type(重量类型)