Changeset 6021 for trunk/puq/get_response.py

Timestamp:

Feb 18, 2016, 4:13:14 PM (9 years ago)

Author:

ldelgass

Message:

Merge UQ and fixes from 1.4 branch

Location:

trunk

Files:

• . (modified) (1 prop)
• puq (copied) (copied from branches/uq/puq)
• puq/get_response.py (copied) (copied from branches/uq/puq/get_response.py) (4 diffs)

trunk/puq/get_response.py

@@ -40 +40 @@
     resp = unpickle(resp)
     var = None
+
     for index, p in enumerate(resp.params):
         if p.name == name:
@@ -50 +51 @@
 
     data = resp.data
-    print ("vars=", resp.vars)
-    print("data=", data)
-    print("my data=", data.T[index])
-    print("my data=", data[:, index])
+    print('data=', repr(data))
+    for ind, p in enumerate(resp.params):
+        if ind == index:
+            continue
+        m = p.pdf.mean
+        means = np.isclose(m, data[:, ind], rtol=1e-6, atol=1e-12)
+        data = data[means]
+
+    print("vars=", resp.vars)
+    print("data=", repr(data))
 
     curve = xml.SubElement(dout, 'curve', {'id': 'response'})
@@ -74 +81 @@
         else:
             allpts[:, i] = np.mean(v.pdf.mean)
-    print("allpts=", allpts)
+
     pts = resp.evala(allpts)
-    print("pts=", pts)
     xy = '\n'.join([' '.join(map(repr, a)) for a in zip(x, pts)])
     comp = xml.SubElement(curve, 'component')
@@ -82 +88 @@
 
     # scatter plot sampled data on response surface
-    # curve = io['output.curve(response-%s-scatter)' % name]
-    # curve['about.label'] = 'Data Points'
-    # curve['about.group'] = title
-    # curve['about.type'] = 'scatter'
-    # curve['xaxis.label'] = varlist[0]
-    # curve['yaxis.label'] = tname
-    # curve['component.xy'] = (xdata, ydata)
+    curve = xml.SubElement(dout, 'curve', {'id': 'scatter'})
+    about = xml.SubElement(curve, 'about')
+    xml.SubElement(about, 'label').text = 'Data Points'
+    xml.SubElement(about, 'group').text = rlabel
+    xml.SubElement(about, 'type').text = 'scatter'
+    comp = xml.SubElement(curve, 'component')
+    xy = '\n'.join([' '.join(map(repr, a)) for a in zip(data[:, index], data[:, -1])])
+    xml.SubElement(comp, 'xy').text = xy