Commit 819de9f4 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Last changes

parent 9cfaee15

deep_chem/utils/evaluate.py

+2 −2
Original line number Diff line number Diff line
@@ -202,7 +202,7 @@ def results_to_csv(results, out, task_type="classification"): 
    if task_type == "classification":

      yscores = np.around(yscores[:,1]).astype(int)

    elif task_type == "regression":

      print yscores

      if type(yscores[0]) == np.ndarray:

        yscores = yscores[:,0]

    with open(out_file, "wb") as csvfile:

      csvwriter = csv.writer(csvfile, delimiter="\t")

CRA Git | Maintained and supported by SUSTech CRA and CCSE