Commit 3cea5427 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

minor debug

parent c447ff2b
...@@ -41,7 +41,6 @@ def hac_iv(diar, scores, method="complete", threshold=0.0): ...@@ -41,7 +41,6 @@ def hac_iv(diar, scores, method="complete", threshold=0.0):
lscores = copy.deepcopy(scores) lscores = copy.deepcopy(scores)
# get the triangular part of the distances # get the triangular part of the distances
distances, t = scores2distance(lscores, threshold) distances, t = scores2distance(lscores, threshold)
# distance = numpy.copy((scores.scoremat + scores.scoremat.T) / 2.0) * -1.0 # distance = numpy.copy((scores.scoremat + scores.scoremat.T) / 2.0) * -1.0
# numpy.fill_diagonal(distance, numpy.inf) # numpy.fill_diagonal(distance, numpy.inf)
# min = numpy.min(distance) # min = numpy.min(distance)
......
...@@ -984,7 +984,7 @@ class Diar(): ...@@ -984,7 +984,7 @@ class Diar():
channel = 'U' channel = 'U'
if diar._attributes.exist('channel'): if diar._attributes.exist('channel'):
channel = segment['channel'] channel = segment['channel']
if time_float: if not time_float:
lst.append('{:s} 1 {:d} {:d} {:s} {:s} {:s} {:s}\n'.format( lst.append('{:s} 1 {:d} {:d} {:s} {:s} {:s} {:s}\n'.format(
segment['show'], segment['start'], segment['stop'] - segment['start'], gender, segment['show'], segment['start'], segment['stop'] - segment['start'], gender,
channel, env, segment['cluster'])) channel, env, segment['cluster']))
......
...@@ -34,7 +34,6 @@ from torch import optim ...@@ -34,7 +34,6 @@ from torch import optim
from torch.utils.data import Dataset from torch.utils.data import Dataset
import logging import logging
from sidekit.nnet.vad_rnn import BLSTM
__license__ = "LGPL" __license__ = "LGPL"
__author__ = "Anthony Larcher" __author__ = "Anthony Larcher"
...@@ -48,7 +47,7 @@ __docformat__ = 'reS' ...@@ -48,7 +47,7 @@ __docformat__ = 'reS'
class PreNet(nn.Module): class PreNet(nn.Module):
def __init(self, def __init__(self,
sample_rate=16000, sample_rate=16000,
windows_duration=0.2, windows_duration=0.2,
frame_shift=0.01): frame_shift=0.01):
...@@ -132,6 +131,7 @@ class BLSTM(nn.Module): ...@@ -132,6 +131,7 @@ class BLSTM(nn.Module):
class SeqToSeq(nn.Module): class SeqToSeq(nn.Module):
def __init__(self): def __init__(self):
super(SeqToSeq, self).__init__()
self.preprocessor = PreNet(sample_rate=16000, self.preprocessor = PreNet(sample_rate=16000,
windows_duration=0.2, windows_duration=0.2,
frame_shift=0.01) frame_shift=0.01)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment