Skip to content

Commit a4bf46e

Browse files
hirosassahirosassa
committed
divide directory
1 parent aabe54c commit a4bf46e

7 files changed

Lines changed: 12 additions & 12 deletions

‎redshells/contrib/train/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from redshells.contrib.train.train_feature_aggregation_similarity_model import TrainFeatureAggregationSimilarityModel
12
from redshells.contrib.train.train_graph_convolutional_matrix_completion import TrainGraphConvolutionalMatrixCompletion
23
from redshells.contrib.train.train_matrix_factorization import TrainMatrixFactorization
34
import redshells.train.utils

redshells/train/train_feature_aggregation_similarity_model.py renamed to redshells/contrib/train/train_feature_aggregation_similarity_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55

66
import gokart
77

8-
from redshells.model import FeatureAggregationSimilarityModel
9-
from redshells.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
8+
from redshells.contrib.model import FeatureAggregationSimilarityModel
9+
from redshells.contrib.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
1010

1111
logger = getLogger(__name__)
1212

‎redshells/contrib/train/train_graph_convolutional_matrix_completion.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
import sklearn
66

77
import gokart
8-
from redshells.model.contrib.gcmc_dataset import GcmcDataset, GcmcGraphDataset
9-
from redshells.model.contrib.graph_convolutional_matrix_completion import GraphConvolutionalMatrixCompletion
8+
from redshells.contrib.model.gcmc_dataset import GcmcDataset, GcmcGraphDataset
9+
from redshells.contrib.model.graph_convolutional_matrix_completion import GraphConvolutionalMatrixCompletion
1010

1111
logger = getLogger(__name__)
1212

‎redshells/train/__init__.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from redshells.train.train_doc2vec import TrainDoc2Vec
66
from redshells.train.train_factorization_machine import TrainFactorizationMachine
77
from redshells.train.train_fasttext import TrainFastText
8-
from redshells.train.train_feature_aggregation_similarity_model import TrainFeatureAggregationSimilarityModel
98
from redshells.train.train_lda_model import TrainLdaModel
109
from redshells.train.train_pairwise_similarity_model import TrainPairwiseSimilarityModel
1110
from redshells.train.train_tfidf import TrainTfidf

‎redshells/train/train_fasttext.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import luigi
77

88
import gokart
9-
import redshells
9+
from redshells.train.utils import TokenIterator
1010

1111
logger = getLogger(__name__)
1212

@@ -31,7 +31,7 @@ def run(self):
3131
shuffle(texts)
3232

3333
if isinstance(texts[0], str):
34-
texts = redshells.train.utils.TokenIterator(texts=texts)
34+
texts = TokenIterator(texts=texts)
3535

3636
logger.info(f'training FastText...')
3737
model = gensim.models.FastText(sentences=texts, **self.fasttext_kwargs)

test/model/test_feature_aggregation_similarity_model.py renamed to test/contrib/model/test_feature_aggregation_similarity_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33

44
import numpy as np
55

6-
from redshells.model import FeatureAggregationSimilarityModel
7-
from redshells.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
6+
from redshells.contrib.model import FeatureAggregationSimilarityModel
7+
from redshells.contrib.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
88

99
logger = getLogger(__name__)
1010

test/train/test_train_feature_aggregation_similarity_model.py renamed to test/contrib/train/test_train_feature_aggregation_similarity_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66

77
import gokart
88

9-
from redshells.model import FeatureAggregationSimilarityModel
10-
from redshells.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
11-
from redshells.train import TrainFeatureAggregationSimilarityModel
9+
from redshells.contrib.model import FeatureAggregationSimilarityModel
10+
from redshells.contrib.model.feature_aggregation_similarity_model import FeatureAggregationSimilarityDataset
11+
from redshells.contrib.train import TrainFeatureAggregationSimilarityModel
1212

1313
logger = getLogger(__name__)
1414

0 commit comments

Comments
 (0)