-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathconfig.py
31 lines (25 loc) · 853 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
"""
Configuration describing the specific problem.
"""
# Author: Arturs Berzins <berzins@cats.rwth-aachen.de>
# License: BSD 3 clause
# For core functionality
datasets = ['train', 'validate', 'test', 'test_Hesthaven_Ubbiali']
num_snapshots = {
'train': 100,
'validate': 50,
'test': 50,
'test_Hesthaven_Ubbiali': 3,
}
# We treat velocity (u,v) as a single vector component u.
components = ['u', 'p']
num_basis = {'u': 30, 'p': 30, }
# A mask for the snapshots in the dataset describing how the components are
# located in the single smallest pattern. Here we treat the velocity vector
# as a single component, so we mask both u and v with 'u'.
mask = ['u','u','p']
# For plotting
import numpy as np
mu_names = ['lx', 'ly', 'angle']
mu_range = {'lx': (1,2), 'ly': (1,2), 'angle': (np.pi/6, 5*np.pi/6)}
P = len(mu_names)