Commit 0973aad4 authored by miaecle's avatar miaecle
Browse files

recover changes

parent e1b925e4
Loading
Loading
Loading
Loading
+3 −2
Original line number Diff line number Diff line
@@ -333,8 +333,9 @@ class TensorGraph(Model):
        feed_dict[self._training_placeholder] = 0.0
        feed_results = self.session.run(outputs, feed_dict=feed_dict)
        if len(feed_results) > 1:
          result = undo_transforms(np.stack(feed_results, 1), transformers)
          feed_results = [result[:, i] for i in range(result.shape[1])]
          if len(transformers):
            raise ValueError("Does not support transformations "
                             "for multiple outputs.")  
        elif len(feed_results) == 1:
          result = undo_transforms(feed_results[0], transformers)
          feed_results = [result]
+1 −6
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ Utility script to convert the benchmark markdown table into a CSV
"""
import sys


def classification_table_to_csv(lines):
  output = []
  headers = [
@@ -28,9 +29,6 @@ def classification_table_to_csv(lines):
    output.append(",".join([split, dataset, model, train, test]))
  for l in output:
    print(l)
  #with open('desired_results.csv', 'a') as f:
  #  for l in output:
  #    f.write(l + '\n')


def regression_table_to_csv(lines):
@@ -57,9 +55,6 @@ def regression_table_to_csv(lines):
    output.append(",".join([split, dataset, model, train, test]))
  for l in output:
    print(l)
  #with open('desired_results.csv', 'a') as f:
  #  for l in output:
  #    f.write(l + '\n')


def split_classification_regression(lines):