simple_train.py

来自「一个功能强大的神经网络分析程序」· Python 代码 · 共 21 行

#!/usr/bin/python
import fann

connection_rate = 1
learning_rate = 0.7
num_input = 2
num_neurons_hidden = 4
num_output = 1

desired_error = 0.0001
max_iterations = 100000
iterations_between_reports = 1000

ann = fann.create(connection_rate, learning_rate, (num_input, num_neurons_hidden, num_output))
ann.set_activation_function_output(fann.SIGMOID_SYMMETRIC_STEPWISE)

ann.train_on_file("datasets/xor.data", max_iterations, iterations_between_reports, desired_error)

ann.save("xor_float.net")