add write_population

Author: AlexLiuyuren

example/client.jl

@@ -23,19 +23,9 @@ par = Parameter(budget=400, evaluation_server_num=2, control_server_ip_port="192
 sol = zoo_min(obj, par)
 # print the Solution object
 sol_print(sol)
-f = open("result.txt", "w")
 positive_data = take!(par.positive_data)
 negative_data = take!(par.negative_data)
-write(f, "########################################\n")
-write(f, "positive_data: \n")
-for sol in positive_data
-    sol_write(sol, f)
-end
-write(f, "########################################\n")
-write(f, "negative_data: \n")
-for sol in negative_data
-    sol_write(sol, f)
-end
+write_population("population.txt", positive_data, negative_data)
 # visualize the optimization progress
 # history = get_history_bestsofar(obj)
 # plt[:plot](history)

src/ZOOclient.jl

@@ -6,7 +6,7 @@ export Dimension, dim_print
 export Objective, get_history_bestsofar
 export zoo_min
 export Parameter, print_population, get_positive_data, get_negative_data
-export Solution, sol_print, sol_equal, sol_write
+export Solution, sol_print, sol_equal, sol_write, write_population
 
 export zoolog
 export rng, my_precision, set_seed, set_precision

src/ZOOclient/algos/asracos/asracos_opt.jl

@@ -132,6 +132,7 @@ function asracos_updater!(asracos::ASRacos, budget, ub, finish)
         f = open(output_file, "w")
     end
     br = false
+
     while(t <= budget)
         sol = take!(asracos.result_set)
         push!(history, sol.value)

src/ZOOclient/parameter.jl

@@ -33,6 +33,7 @@ type Parameter
     # result
     positive_data
     negative_data
+
     function Parameter(; algorithm="asracos", budget=0, init_sample=Nullable(),
         time_limit=Nullable{Int64}(), precision=Nullable(),
         uncertain_bits=Nullable{Int64}(), train_size=0, positive_size=0, negative_size=0,
@@ -84,19 +85,3 @@ function get_negative_data(parameter)
     result = take!(parameter.negative_data)
     return result
 end
-
-function write_population(parameter, file)
-    f = open(file, "w")
-    positive_data = get_positive_data(parameter)
-    negative_data = get_negative_data(parameter)
-    write(f, "########################################\n")
-    write(f, "positive_data: \n")
-    for sol in positive_data
-        sol_write(sol, f)
-    end
-    write(f, "########################################\n")
-    write(f, "negative_data: \n")
-    for sol in negative_data
-        sol_write(sol, f)
-    end
-end

src/ZOOclient/solution.jl

@@ -63,7 +63,22 @@ function sol_print(sol)
     zoolog("x: $(sol.x)")
 end
 
-function sol_write(sol, f)
+function sol_write(f, sol)
     write(f, "value=$(sol.value)\n")
     write(f, "x=$(sol.x)\n")
+end
+
+function write_population(filename, positive_data, negative_data)
+	f = open(filename, "w")
+	write(f, "########################################\n")
+	write(f, "positive_data: \n")
+	for sol in positive_data
+		sol_write(f, sol)
+	end
+	write(f, "########################################\n")
+	write(f, "negative_data: \n")
+	for sol in negative_data
+		sol_write(f, sol)
+	end
+	close(f)
 end