0e3625de80
* equation class * difference domain * dummy dataloader * writer class * refactoring and minor fix
27 lines
693 B
Python
27 lines
693 B
Python
"""Module for Location class."""
|
|
|
|
from .location import Location
|
|
from ..label_tensor import LabelTensor
|
|
|
|
class Difference(Location):
|
|
"""
|
|
"""
|
|
def __init__(self, first, second):
|
|
|
|
|
|
self.first = first
|
|
self.second = second
|
|
|
|
def sample(self, n, mode ='random', variables='all'):
|
|
"""
|
|
"""
|
|
assert mode is 'random', 'Only random mode is implemented'
|
|
|
|
samples = []
|
|
while len(samples) < n:
|
|
sample = self.first.sample(1, 'random')
|
|
if not self.second.is_inside(sample):
|
|
samples.append(sample.tolist()[0])
|
|
|
|
import torch
|
|
return LabelTensor(torch.tensor(samples), labels=['x', 'y']) |