SWE/Object/Repository.pm

package SWE::Object::Repository;
use strict;
use warnings;

sub new ($%) {
  my $class = shift;
  my $self = bless {@_}, $class;

  return $self;
}

sub db ($) { $_[0]->{db} }

my $weight_file_name = 'data/weight.txt';

sub term_weight_vector ($) {
  my $self = shift;

  ## TODO: lock

  ## TODO: use global props

  return $self->{term_weight_vector} ||= do {
    require SWE::Data::FeatureVector;

    my $w;
    if (-f $weight_file_name) {
      local $/ = undef;
      open my $file, '<:encoding(utf8)', $weight_file_name or die "$0: $weight_file_name: $!";
      $w = SWE::Data::FeatureVector->parse_stringref (\<$file>);
      close $file;
    } else {
      $w = SWE::Data::FeatureVector->new;
    }
    delete $self->{term_weight_vector_modified};
    $w;
  };
} # term_weight_vector

sub save_term_weight_vector ($) {
  my $self = shift;
  return unless $self->{term_weight_vector_modified};
  
  ## TODO: use global props
  
  open my $file, '>:encoding(utf8)', $weight_file_name or die "$0: $weight_file_name: $!";
  print $file $self->{term_weight_vector}->stringify;
  close $file;
} # save_term_weight_vector

sub are_related_ids ($$$;$) {
  my ($self, $id1, $id2, $answer) = @_;
  
  my $w = $self->term_weight_vector;
  
  my $tfidf_db = $self->db->id_tfidf;
  
  require SWE::Data::FeatureVector;
  my $fv1 = SWE::Data::FeatureVector->parse_stringref
      ($tfidf_db->get_data ($id1));
  my $fv2 = SWE::Data::FeatureVector->parse_stringref
      ($tfidf_db->get_data ($id2));
  
  my $diff = $fv1->subtract ($fv2);
  
  my $wx = $diff->multiply ($w)->component_sum;
  my $y = $wx >= 0 ? 1 : -1;
  
  if (defined $answer and $y * $answer < 0) {
    my $new_w = $y > 0 ? $w->subtract ($diff) : $w->add ($diff);
    ## TODO: while
    $self->{term_weight_vector} = $new_w;
    $self->{term_weight_vector_modified} = 1;
  }

  return $y > 0;
} # are_related_ids

1;
1	wakaba	1.1	package SWE::Object::Repository;
2			use strict;
3			use warnings;
4
5			sub new ($%) {
6			my $class = shift;
7			my $self = bless {@_}, $class;
8
9			return $self;
10			}
11
12			sub db ($) { $_[0]->{db} }
13
14			my $weight_file_name = 'data/weight.txt';
15
16			sub term_weight_vector ($) {
17			my $self = shift;
18
19			## TODO: lock
20
21			## TODO: use global props
22
23			return $self->{term_weight_vector} \|\|= do {
24			require SWE::Data::FeatureVector;
25
26			my $w;
27			if (-f $weight_file_name) {
28			local $/ = undef;
29			open my $file, '<:encoding(utf8)', $weight_file_name or die "$0: $weight_file_name: $!";
30			$w = SWE::Data::FeatureVector->parse_stringref (\<$file>);
31			close $file;
32			} else {
33			$w = SWE::Data::FeatureVector->new;
34			}
35			delete $self->{term_weight_vector_modified};
36			$w;
37			};
38			} # term_weight_vector
39
40			sub save_term_weight_vector ($) {
41			my $self = shift;
42			return unless $self->{term_weight_vector_modified};
43
44			## TODO: use global props
45
46			open my $file, '>:encoding(utf8)', $weight_file_name or die "$0: $weight_file_name: $!";
47			print $file $self->{term_weight_vector}->stringify;
48			close $file;
49			} # save_term_weight_vector
50
51			sub are_related_ids ($$$;$) {
52			my ($self, $id1, $id2, $answer) = @_;
53
54			my $w = $self->term_weight_vector;
55
56			my $tfidf_db = $self->db->id_tfidf;
57
58			require SWE::Data::FeatureVector;
59			my $fv1 = SWE::Data::FeatureVector->parse_stringref
60			($tfidf_db->get_data ($id1));
61			my $fv2 = SWE::Data::FeatureVector->parse_stringref
62			($tfidf_db->get_data ($id2));
63
64			my $diff = $fv1->subtract ($fv2);
65
66			my $wx = $diff->multiply ($w)->component_sum;
67			my $y = $wx >= 0 ? 1 : -1;
68
69			if (defined $answer and $y * $answer < 0) {
70			my $new_w = $y > 0 ? $w->subtract ($diff) : $w->add ($diff);
71			## TODO: while
72			$self->{term_weight_vector} = $new_w;
73			$self->{term_weight_vector_modified} = 1;
74			}
75
76			return $y > 0;
77			} # are_related_ids
78
79			1;